TopFormer
TopFormer copied to clipboard
Is batchsize>1 supported in convert2onnx.py file?
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:
And perform model conversion script execution:
But the generated model is still batchsize=1:
What do I need to achieve onnx model generation with batchsize=2?
Thank you very much, looking forward to your reply!
Your question is heavily related to the MMSegmentation framework and ONNX. It could be better to raise an issue under the corresponding repositories.