Skip to content

Commit

Permalink
Update test_cpu.py
Browse files Browse the repository at this point in the history
  • Loading branch information
williamFalcon authored and awaelchli committed May 20, 2020
1 parent 20bcd68 commit b0245e2
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/models/test_cpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -355,4 +355,3 @@ def test_auto_move_data(tmpdir):
assert model(x).device == torch.device('cpu'), "Automoving data to same device as model failed"
x = x.cuda(0)
assert model(x).device == torch.device('cpu'), "Automoving data to same device as model failed"

0 comments on commit b0245e2

Please sign in to comment.