main code

update trace
sbing2000 · Jul 20, 2022 · 1e1e0d5 · 1e1e0d5
1 parent 8b72ac7
commit 1e1e0d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/yolo.py b/models/yolo.py
@@ -437,7 +437,7 @@ def forward_once(self, x, profile=False):
                 self.traced=False
 
             if self.traced:
-                if isinstance(m, Detect) or isinstance(m, IDetect) or isinstance(m, IAuxDetect):
+                if isinstance(m, Detect) or isinstance(m, IDetect) or isinstance(m, IAuxDetect) or isinstance(m, IKeypoint):
                     break
 
             if profile: