Skip to content

Commit d059d1d

Browse files
authored
Report fused model summary by default (#7722)
1 parent aa7a0e9 commit d059d1d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

models/yolo.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,3 +333,6 @@ def parse_model(d, ch): # model_dict, input_channels(3)
333333
_ = Model(cfg)
334334
except Exception as e:
335335
print(f'Error in {cfg}: {e}')
336+
337+
else: # report fused model summary
338+
model.fuse()

0 commit comments

Comments
 (0)