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()
|