facebookresearch / facebookresearch/fvcore
Add device parameter to update_bn_stats() in precise_bn.py
Open
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 236
- PR merge metrics
- No merged PRs in 30d
Description
Hello. Thank you for sharing your work!
I want to use precise_bn, but there is no device option in `update_bn_stats()`. Can we add device option in `update_bn_stats()`?
https://github.com/facebookresearch/fvcore/blob/c1ae9e7173259fc92d24b05b49558787c457b76a/fvcore/nn/precise_bn.py#L88
Again, thank you.
Contributor guide
Assessment
This issue has not been assessed yet.