TopFormer icon indicating copy to clipboard operation
TopFormer copied to clipboard

Is batchsize>1 supported in convert2onnx.py file?

Open tensorflowt opened this issue 2 years ago • 1 comments

Hi! Thank you so much for your open source project, it's great! I encountered a problem when verifying its performance, as follows: When I want to test batchsize=2, convert the pth model to the onnx model. The specific operations are as follows: 864cd9253c929a44f7e5ba3ae149160 And perform model conversion script execution: 434eec985937b597c56e6ceb9aff0f0 But the generated model is still batchsize=1: 0e883d8c29cfd465294fe13deef078f What do I need to achieve onnx model generation with batchsize=2? Thank you very much, looking forward to your reply!

tensorflowt avatar May 16 '22 09:05 tensorflowt

Your question is heavily related to the MMSegmentation framework and ONNX. It could be better to raise an issue under the corresponding repositories.

speedinghzl avatar May 30 '22 13:05 speedinghzl