导出onnx模型时出错。

Question

下面是模型：SMSR

当我试图使用torch.onnx.export(self.model、lr、onnx_save_path、verbose=True)导出onnx模型时，会发生以下错误：

TyperError: state_dict()采用一个位置参数，但给出了3个

我的火炬版本是: 1.1.0

那我该怎么办？

谢谢大家！

Answer 1

Torch版本已经很老了，您的代码将运行在1.12版本中。