Skip to content

Commit

Permalink
Report fused model summary by default (#7722)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed May 7, 2022
1 parent aa7a0e9 commit d059d1d
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 d059d1d

Please sign in to comment.