utils/torch_utils.py · 6aed0a7c00c955383cc04a1d9210898b654c700e · Administrator / yolov5

GFLOPs computation fix for classification models (#8954) · 6aed0a7c

由提交于 8月 13, 2022

* GFLOPs computation fix for classification models

Improved robustness in reading input channel count

* Update torch_utils.py

* Update torch_utils.py

6aed0a7c

torch_utils.py 16.8 KB

Replace torch_utils.py