Remove unwanted .cuda()

harvardnlp · Oct 1, 2021 · e65832e · e65832e
1 parent acbce53
commit e65832e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_alignment_crf.py b/tests/test_alignment_crf.py
@@ -5,7 +5,7 @@
 
 def test_alignment_crf():
     batch, N, M = 1, 4, 5
-    log_potentials = torch.rand(batch, N, M, 3).cuda()
+    log_potentials = torch.rand(batch, N, M, 3)
 
     if torch.cuda.is_available():
         log_potentials = log_potentials.cuda()