How to output model size information？

iscyy / yoloair

🔥🔥🔥 专注于YOLOv5，YOLOv7、YOLOv8、YOLOv9改进模型，Support to improve backbone, neck, head, loss, IoU, NMS and other modules🚀

https://github.com/iscyy/yoloair

GNU General Public License v3.0

2.47k stars 428 forks source link

Closed Grenaf closed 1 year ago

Grenaf commented 1 year ago

When I used YOLOair‘s train.py the model size information only included 369 layers, 7514547 parameters, 7514547 gradients，just like this

Is there a way to get BFLOPS and FPS information？

problem has been sloved！ pip install thop