用户
 找回密码
 立即注册
发表于 2021-7-2 18:02:21
13360
我想在TRT中使用动态batch和动态输入,但是在网络中存在全连接层就必须在进入网络之前进行一次resize操作,
那么addinput时候设置了动态batch和动态输入(nchw);
在其之后增加了一个addresize的操作,并规定addresize设置输出维度和input维度一致,那么岂不是addresize会将batch和chw都固定吗?
我现在只想固定chw,而动态batch,请问这个操作如何实现?

使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册