Closed SMKamrulHasan closed 5 years ago
File "train.py", line 114, in target = target.type(torch.FloatTensor) NameError: name 'target' is not defined
You need to run it with --cuda. CUDA_VISIBLE_DEVICES=<your_gpu_id> python train.py --cuda
CUDA_VISIBLE_DEVICES=<your_gpu_id> python train.py --cuda
File "train.py", line 114, in
target = target.type(torch.FloatTensor)
NameError: name 'target' is not defined