Commit 46d932ae authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed bug in model.test_if_cuda

parent 581f83b5
......@@ -176,7 +176,7 @@ class Solver():
X = torch.unsqueeze(X, dim= 1)
y = torch.unsqueeze(y, dim= 1)
if model.test_if_cuda():
if model.test_if_cuda:
X = X.cuda(self.device, non_blocking=True)
y = y.cuda(self.device, non_blocking=True)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment