Fix export to export same output as detect

robin-maillot · Nov 2, 2020 · c4200bc · c4200bc
1 parent 10a1df7
commit c4200bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/export.py b/models/export.py
@@ -47,7 +47,7 @@
             m.act = Hardswish()  # assign activation
         # if isinstance(m, models.yolo.Detect):
         #     m.forward = m.forward_export  # assign forward (optional)
-    model.model[-1].export = True  # set Detect() layer export=True
+    # model.model[-1].export = True  # set Detect() layer export=True
     y = model(img)  # dry run
 
     # TorchScript export