gwennlbh commited on
Commit
71f3de2
·
verified ·
1 Parent(s): 08386cc

Fix andrena classifier: put external data into the onnx file

Browse files
Files changed (2) hide show
  1. classifier-andrena.onnx +2 -2
  2. pytorch-to-onnx.py +1 -0
classifier-andrena.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c891ce12e38af92df1a0bdf75a43db93132f14e4eba7e1b52f02c32fc5091492
3
- size 1063717
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a9ae2bc8fa555c22113409bfda9d104d0e7deb36e0121ef0f689550531fc5c9
3
+ size 112665804
pytorch-to-onnx.py CHANGED
@@ -50,4 +50,5 @@ torch.onnx.export(
50
  base_model,
51
  args=(torch.zeros([1, 3, image_input_size, image_input_size]),),
52
  f=filename.with_suffix(".onnx"),
 
53
  )
 
50
  base_model,
51
  args=(torch.zeros([1, 3, image_input_size, image_input_size]),),
52
  f=filename.with_suffix(".onnx"),
53
+ external_data=False
54
  )