utils · 05a955a3f6a580307f744178a9ad6938cb326096 · Administrator / yolov5

FLOPS computation device bug fix (#1447) · 05a955a3

由提交于 11月 19, 2020

* Update torch_utils.py

fix issue#113 , inputs device should be same with model parameters' device

* Update torch_utils.py

* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

05a955a3

名称	最后提交	最后更新
..
google_app_engine		正在载入提交数据...
__init__.py		正在载入提交数据...
activations.py		正在载入提交数据...
autoanchor.py		正在载入提交数据...
datasets.py		正在载入提交数据...
general.py		正在载入提交数据...
google_utils.py		正在载入提交数据...
loss.py		正在载入提交数据...
metrics.py		正在载入提交数据...
plots.py		正在载入提交数据...
torch_utils.py		正在载入提交数据...