Open anitchakraborty opened 4 years ago
You are not using the CUDA version of PyTorch. Try passing in an argument like python ./src/train.py --device cpu
, or, if you have a NVIDIA GPU, reinstall the CUDA version of PyTorch following https://pytorch.org/get-started/locally/
While training the model via train.py file in the task 3/src/train.py, the following error is popping up, any assistance shall be immensely helpful.