Skip to content

Commit

Permalink
Report fused model summary by default (ultralytics#7722)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher authored and Clay Januhowski committed Sep 8, 2022
1 parent 33116c5 commit bc669fc
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions models/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

0 comments on commit bc669fc

Please sign in to comment.