Browse Source

Report fused model summary by default (#7722)

modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
d059d1da03
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      models/yolo.py

+ 3
- 0
models/yolo.py View File

@@ -333,3 +333,6 @@ if __name__ == '__main__':
_ = Model(cfg)
except Exception as e:
print(f'Error in {cfg}: {e}')

else: # report fused model summary
model.fuse()

Loading…
Cancel
Save