apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

main code

Browse Source 
update trace
This commit is contained in:
Kin-Yiu, Wong 2022-07-20 14:46:12 +08:00 committed by GitHub
parent 8b72ac7792
commit 1e1e0d52d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/yolo.py
View File

@ -437,7 +437,7 @@ class Model(nn.Module):
self.traced=False self.traced=False
if self.traced: 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 break
if profile: if profile: