Open StrugglingForBetter opened 4 years ago
File "train.py", line 260, in mask = torch.from_numpy(mask).to(DEVICE) RuntimeError: CUDA error: an illegal memory access was encountered
File "train.py", line 260, in
mask = torch.from_numpy(mask).to(DEVICE)
RuntimeError: CUDA error: an illegal memory access was encountered