Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Andrei-Claudiu Roibu
BrainMapper
Commits
46d932ae
Commit
46d932ae
authored
Apr 16, 2020
by
Andrei-Claudiu Roibu
🖥
Browse files
fixed bug in model.test_if_cuda
parent
581f83b5
Changes
1
Hide whitespace changes
Inline
Side-by-side
solver.py
View file @
46d932ae
...
...
@@ -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
)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment