jlin1212 opened this issue 3 years ago · 0 comments
On lines 17 and 73, torch.cuda.is_available should instead be torch.cuda.is_available().
torch.cuda.is_available
torch.cuda.is_available()