pythonL123

请问这个问题解决了吗 我也出现这个问题

噢噢 好的 谢谢

转export的代码是 model = Siamese(input_shape=[35, 35], pretrained=False) model.load_state_dict(torch.load('logs/ep500-loss0.181-val_loss0.023.pth')) model.eval() dummy_input = torch.randn(1, 3, 35, 35) output_path = "model.onnx" torch.onnx.export(model, dummy_input, output_path, export_params=True, do_constant_folding=True, input_names=['input'], output_names=['output']) 然后报错 File "/home/lium/Siamese-pytorch-master/nets/siamese.py", line 31, in...