MCG-NJU/VideoMAE

AttributeError: module 'torch._C' has no attribute '_get_privateuse1_backend_name'

abhisheksushil2003 opened this issue · 0 comments

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():