From f9633c81ed0fae58493526ad6ebeae3b92417f24 Mon Sep 17 00:00:00 2001
From: Raul James <raul@nobi.io>
Date: Tue, 24 Jan 2023 22:32:29 +0100
Subject: [PATCH] Fix export to get rid of symbolic dimension in output

use torch=1.7.1 instead
see https://github.com/ultralytics/yolov5/issues/2558, maybe related
---
 requirements.txt          | 4 ++--
 yolox-repo/yolox.prototxt | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 9ccb3f8..e5e1264 100644
--- 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
index f1811f2..2adcb71 100644
--- 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
   }