Fix andrena classifier: put external data into the onnx file
Browse files- classifier-andrena.onnx +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:
|
| 3 |
-
size
|
|
|
|
| 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 |
)
|