Browse Source

Update export.py

master
bailulu 3 years ago
parent
commit
3959591071
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      export.py

+ 1
- 1
export.py View File

@@ -62,7 +62,7 @@ if __name__ == '__main__':
if opt.include_nms: if opt.include_nms:
model.model[-1].include_nms = True model.model[-1].include_nms = True
y = None y = None
# TorchScript export 1
# TorchScript export 11
try: 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.pt') # filename f = opt.weights.replace('.pt', '.torchscript.pt') # filename


Loading…
Cancel
Save