NVlabs / SegFormer

Official PyTorch implementation of SegFormer
https://arxiv.org/abs/2105.15203
Other
2.58k stars 357 forks source link

RuntimeError: Only tuples, lists and Variables supported as JIT inputs/outputs. Dictionaries and strings are also accepted but their usage is not recommended. But got unsupported type numpy.ndarray #140

Closed hengyanchen closed 11 months ago

hengyanchen commented 11 months ago

There's a error message,when i use pytorch2onnx.py to convert a onnx model batchsize>1 : "RuntimeError: Only tuples, lists and Variables supported as JIT inputs/outputs. Dictionaries and strings are also accepted but their usage is not recommended. But got unsupported type numpy.ndarray"

But it's can be convert sucessful when batchsize=1