Commit 581f83b5 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed model.is_cuda bug

parent 29266e50
......@@ -176,7 +176,7 @@ class Solver():
X = torch.unsqueeze(X, dim= 1)
y = torch.unsqueeze(y, dim= 1)
if model.is_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