Pytorch to onnx fail

[jasper2326]

RuntimeError: Only tuples, lists and Variables are supported as JIT inputs/outputs. Dictionaries and strings are also accepted, but their usage is not recommended. Here, received an input of unsupported type: numpy.ndarray

Can anyone help me with this issue? Thanks.

[chenzx2]

I got the same problem

[chenzx2]

here is my env: ubantu18 torch 2.0.0+cu117 torchaudio 2.0.1+cu117 mmcv 1.7.0 mmcv-full 1.7.0

[chenzx2]

RuntimeError: Only tuples, lists and Variables are supported as JIT inputs/outputs. Dictionaries and strings are also accepted, but their usage is not recommended. Here, received an input of unsupported type: numpy.ndarray

Can anyone help me with this issue? Thanks.

here is my change

one_meta = {
    'img_shape': (H, W, C),
    'ori_shape': (H, W, C),
    'pad_shape': (H, W, C),
    'filename': '<demo>.png',
    'scale_factor': torch.from_numpy(np.ones(4, dtype=np.float32) ),
    'flip': False,
    'show_img': torch.from_numpy(show_img),
    'flip_direction': None
}

[shuyue-jl]

I got the same problem