facebookresearch / facebookresearch/SlowFast
Use PyTorch core for some utils in utils/distributed.py
Open
- Dominant language
- Python
- Stars
- 7.4k
- Forks
- 1.3k
- PR merge metrics
- No merged PRs in 30d
Description
Similar to issue https://github.com/facebookresearch/detectron2/issues/2750 filed in detectron repo.
The distributed utils file over at https://github.com/facebookresearch/SlowFast/blob/master/slowfast/utils/distributed.py has some APIs to perform all_gather, gather, etc on arbitrary pickable data. As of 1.8, PyTorch now provides these features natively, so it would be great to move these utils to use those APIs so that we can eliminate unnecessary code and benefit from upstream PyTorch improvements.
Happy to send the PR as needed .
Contributor guide
Assessment
This issue has not been assessed yet.