AttributeError: module 'torch._C' has no attribute '_get_privateuse1_backend_name'
abhisheksushil2003 opened this issue · 0 comments
abhisheksushil2003 commented
When running the code for pretraining on kinetics data set, without distributing it across multiple nodes. I am getting the following error.
AttributeError: module 'torch._C' has no attribute '_get_privateuse1_backend_name'
from this line in the dataloader.py file.
elif self._pin_memory_device == torch._C._get_privateuse1_backend_name():