Commit
•
d059d1d
1
Parent(s):
aa7a0e9
Report fused model summary by default (#7722)
Browse files- models/yolo.py +3 -0
models/yolo.py
CHANGED
|
@@ -333,3 +333,6 @@ if __name__ == '__main__':
|
|
| 333 |
_ = Model(cfg)
|
| 334 |
except Exception as e:
|
| 335 |
print(f'Error in {cfg}: {e}')
|
|
|
|
|
|
|
|
|
|
|
|
| 333 |
_ = Model(cfg)
|
| 334 |
except Exception as e:
|
| 335 |
print(f'Error in {cfg}: {e}')
|
| 336 |
+
|
| 337 |
+
else: # report fused model summary
|
| 338 |
+
model.fuse()
|