diff --git a/models/yolo.py b/models/yolo.py index f072aeeb8eac..b17a59c376f6 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -333,3 +333,6 @@ def parse_model(d, ch): # model_dict, input_channels(3) _ = Model(cfg) except Exception as e: print(f'Error in {cfg}: {e}') + + else: # report fused model summary + model.fuse()