Skip to content
Snippets Groups Projects
Commit 9bb982de authored by Franck Galpin's avatar Franck Galpin
Browse files

correct bad parameters in to_onnx conversion

parent 4d7a1439
No related branches found
No related tags found
No related merge requests found
......@@ -68,4 +68,4 @@ modelobj = Trainer.instantiate_from_dict(model, config["model"])
checkpoint = torch.load(args.ckpt, map_location=torch.device("cpu"))
modelobj.load_state_dict(checkpoint["model"])
modelobj.SADL_model.to_onnx(args.output, args.batch_size, args.patch_size)
modelobj.SADL_model.to_onnx(args.output, args.patch_size, args.batch_size)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment