Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Patrick Labatut's avatar
    Tidy uses of torch.device in Volumes · 1f9661e1
    Patrick Labatut authored 
    Summary:
Tidy uses of `torch.device` in `Volumes`:
- Allow `str` or `torch.device` in `Volumes.to()` method
- Consistently use `torch.device` for internal type
- Fix comparison of devices

Reviewed By: nikhilaravi

Differential Revision: D28970876

fbshipit-source-id: c640cc22ced684a54cc450ac38a0f4b3435d47be
    1f9661e1
test_volumes.py 32.6 KB