Fix export to get rid of symbolic dimension in output

use torch=1.7.1 instead see ultralytics/yolov5#2558, maybe related
edgeimpulse · Jan 24, 2023 · f9633c8 · f9633c8
1 parent 02f384a
commit f9633c8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ flatbuffers==1.12
 onnx==1.11.0
 sklearn==0.0
 tensorflow_probability==0.15.0 # onnx-tf requirement (TF2.7 compatibility)
-torch==1.11.0 # why does no-one pin their dependencies, 1.12 does not work
-torchvision==0.12.0 # pairs w/ pytorch 1.11.0
+torch==1.7.1 # https://github.com/ultralytics/yolov5/issues/2558
+torchvision==0.8.2
 Pillow==9.0.0
 pycocotools==2.0.5
diff --git a/yolox-repo/yolox.prototxt b/yolox-repo/yolox.prototxt
@@ -1,17 +1,17 @@
 name: "yolo_v3"
 tidl_yolo {
   yolo_param {
-    input: "onnx::Slice_709"
+    input: "709"
     anchor_width: 8.0
     anchor_height: 8.0
   }
   yolo_param {
-    input: "onnx::Slice_794"
+    input: "843"
     anchor_width: 16.0
     anchor_height: 16.0
   }
   yolo_param {
-    input: "output"
+    input: "977"
     anchor_width: 32.0
     anchor_height: 32.0
   }