main code

update trace
2022-07-20 14:46:12 +08:00 · 2022-07-20 14:46:12 +08:00 · 1e1e0d52d9
parent 8b72ac7792
commit 1e1e0d52d9
1 changed files with 1 additions and 1 deletions
--- a/models/yolo.py
+++ b/models/yolo.py
@ -437,7 +437,7 @@ class Model(nn.Module):
                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: