LittleQQ

convert to onnx

1

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...