From a9d20eba3e61ac5b7e77e0a7c07828fb2cf84bb5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 3 Jul 2020 11:50:59 -0700 Subject: [PATCH] export.py update --- models/export.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/models/export.py b/models/export.py index 42fd577..1ff9526 100644 --- a/models/export.py +++ b/models/export.py @@ -30,7 +30,7 @@ if __name__ == '__main__': # TorchScript export try: - print('\nStarting TorchScript export with torch %s...' % torch.__version) + print('\nStarting TorchScript export with torch %s...' % torch.__version__) f = opt.weights.replace('.pt', '.torchscript') # filename ts = torch.jit.trace(model, img) ts.save(f) @@ -52,6 +52,9 @@ if __name__ == '__main__': onnx_model = onnx.load(f) # load onnx model onnx.checker.check_model(onnx_model) # check onnx model print(onnx.helper.printable_graph(onnx_model.graph)) # print a human readable model - print('ONNX export success, saved as %s\nView with https://github.com/lutzroeder/netron' % f) + print('ONNX export success, saved as %s' % f) except Exception as e: print('ONNX export failed: %s' % e) + + # Finish + print('\nExports complete. Visualize with https://github.com/lutzroeder/netron.') \ No newline at end of file