fixed test seq2seq

Signed-off-by: Anhforth <[email protected]>
FlagAI-Open · Jun 24, 2022 · 1a4e253 · 1a4e253
1 parent 2d9c786
commit 1a4e253
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_glm_seq2seq.py b/tests/test_glm_seq2seq.py
@@ -25,7 +25,7 @@ def test_init_trainer_pytorch(self):
                           log_interval=50,
                           experiment_name='glm_large',
                           fp16=True,
-                          pytorch_device='cuda:1',
+                          pytorch_device='cuda',
                           lr=1e-4)
         print("downloading...")