convert to onnx

[LittleQQ]

i convert model to onnx with following code: device = 'cuda' if torch.cuda.is_available() else 'cpu' modelpath = './model_final.pt' model = torch.load(modelpath) model.to(device) model.eval() dummy_input = torch.randn(1, 3, 512, 512).to(device) onnx_path = './model_final.onnx' torch.onnx.export(model.module, dummy_input, onnx_path) it can convert successfully but i inference with onnx model ,it can't load onnx model ,how can solve it ?

[pjl1995]

Maybe you can see if the absolute path and file name are wrong.