glenn-jocher commited on
Commit
3ef3a95
·
unverified ·
1 Parent(s): b8be76f

Do not optimize CoreML TorchScript model (#3055)

Browse files
Files changed (1) hide show
  1. models/export.py +1 -2
models/export.py CHANGED
@@ -75,8 +75,7 @@ if __name__ == '__main__':
75
  print(f'\n{prefix} starting export with torch {torch.__version__}...')
76
  f = opt.weights.replace('.pt', '.torchscript.pt') # filename
77
  ts = torch.jit.trace(model, img, strict=False)
78
- ts = optimize_for_mobile(ts) # https://pytorch.org/tutorials/recipes/script_optimized.html
79
- ts.save(f)
80
  print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
81
  except Exception as e:
82
  print(f'{prefix} export failure: {e}')
 
75
  print(f'\n{prefix} starting export with torch {torch.__version__}...')
76
  f = opt.weights.replace('.pt', '.torchscript.pt') # filename
77
  ts = torch.jit.trace(model, img, strict=False)
78
+ optimize_for_mobile(ts).save(f) # https://pytorch.org/tutorials/recipes/script_optimized.html
 
79
  print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
80
  except Exception as e:
81
  print(f'{prefix} export failure: {e}')