From 3a90eaa9b9b32c2883fb60a871928e0ae99e2e62 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 6 May 2022 23:48:04 -0700 Subject: [PATCH] Report fused model summary by default --- models/yolo.py | 3 +++ 1 file changed, 3 insertions(+) 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()