From 7a098eb8cd8ceb73b44c89ebfac3bdb1527fb35b Mon Sep 17 00:00:00 2001 From: bailulu <996950537@qq.com> Date: Sat, 15 Oct 2022 11:37:52 +0800 Subject: [PATCH] Update export.py --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 8fd0d6e..101671e 100644 --- a/export.py +++ b/export.py @@ -62,7 +62,7 @@ if __name__ == '__main__': if opt.include_nms: model.model[-1].include_nms = True y = None - # TorchScript export + # TorchScript export 1 try: print('\nStarting TorchScript export with torch %s...' % torch.__version__) f = opt.weights.replace('.pt', '.torchscript.pt') # filename