Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Report fused model summary by default (#7722)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-05-06 23:51:59 -07:00
1 parent aa7a0e91b1
commit d059d1da03
1 file changed
+3
Download Patch File Download Diff File Expand all files Collapse all files
models/yolo.py
+3
View File
@@ -333,3 +333,6 @@ if __name__ == '__main__':
_ = Model(cfg) _ = Model(cfg)
except Exception as e: except Exception as e:
print(f'Error in {cfg}: {e}') print(f'Error in {cfg}: {e}')
else: # report fused model summary
model.fuse()