我想在TRT中使用动态batch和动态输入,但是在网络中存在全连接层就必须在进入网络之前进行一次resize操作,
那么addinput时候设置了动态batch和动态输入(nchw);
在其之后增加了一个addresize的操作,并规定addresize设置输出维度和input维度一致,那么岂不是addresize会将batch和chw都固定吗?
我现在只想固定chw,而动态batch,请问这个操作如何实现?
我想在TRT中使用动态batch和动态输入,但是在网络中存在全连接层就必须在进入网络之前进行一次resize操作,
那么addinput时候设置了动态batch和动态输入(nchw);
在其之后增加了一个addresize的操作,并规定addresize设置输出维度和input维度一致,那么岂不是addresize会将batch和chw都固定吗?
我现在只想固定chw,而动态batch,请问这个操作如何实现?