export_ONNX报错

当使用torch导出ONNX模型的时候报错如下

for idx, r_split in enumerate(r_splits):

TypeError: ‘torch._C.Value’ object is not iterable
(Occurred when translating repeat_interleave).

解决方法:更新torch到2.0即可


export_ONNX报错
https://johnson7788.github.io/2023/04/19/export-ONNX%E6%8A%A5%E9%94%99/
作者
Johnson
发布于
2023年4月19日
许可协议