diff --git a/models/yolo.py b/models/yolo.py index f072aee..b17a59c 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -333,3 +333,6 @@ if __name__ == '__main__': _ = Model(cfg) except Exception as e: print(f'Error in {cfg}: {e}') + + else: # report fused model summary + model.fuse()