fix ONNX model

export_to_torchscript.py

@@ -97,7 +97,7 @@ torch.onnx.export(
     input_names=["image"],
     output_names=["embedding"],
     dynamic_axes={
-        "input": {0: "batch_size"},
-        "output": {0: "batch_size"},
+        "image": {0: "batch_size"},
+        "embedding": {0: "batch_size"},
     },
 )

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eec3f58a2b57142c48411467858847f75b2726f7b22db877428fcf06aaac2958
-size 112312109
+oid sha256:1db3c0f65f4256133f2d81fe2d1a410c465626cd303b71ee46028da6bf9f118f
+size 112494553