From 1e1e0d52d99f444d8df0262b6e621cf2cab6068b Mon Sep 17 00:00:00 2001
From: "Kin-Yiu, Wong" <102582011@cc.ncu.edu.tw>
Date: Wed, 20 Jul 2022 14:46:12 +0800
Subject: [PATCH] main code

update trace
---
 models/yolo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/yolo.py b/models/yolo.py
index fb7f1eca..dcb073ee 100644
--- 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: