Closed pvmilk closed 6 years ago
You probably forget the parenthesis there
if torch.cuda.is_available(): return n.cuda() <==
You're right! Thanks for finding this error
You probably forget the parenthesis there