Merge pull request PaddlePaddle#22 from LiuChiachi/fix-seq2seq-args-bug

Fix seq2seq args bug, replace use_gpu with select_device
wangxicoding · Feb 20, 2021 · f7bd0ab · f7bd0ab
2 parents 5b35fb9 + 1fa4663
commit f7bd0ab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/machine_translation/seq2seq/predict.py b/examples/machine_translation/seq2seq/predict.py
@@ -41,7 +41,7 @@ def post_process_seq(seq, bos_idx, eos_idx, output_bos=False, output_eos=False):
 
 
 def do_predict(args):
-    device = paddle.set_device("gpu" if args.use_gpu else "cpu")
+    device = paddle.set_device(args.select_device)
 
     test_loader, src_vocab_size, tgt_vocab_size, bos_id, eos_id = create_infer_loader(
         args)

diff --git a/examples/machine_translation/seq2seq/train.py b/examples/machine_translation/seq2seq/train.py
@@ -23,7 +23,7 @@
 
 
 def do_train(args):
-    device = paddle.set_device("gpu" if args.use_gpu else "cpu")
+    device = paddle.set_device(args.select_device)
 
     # Define dataloader
     train_loader, eval_loader, src_vocab_size, tgt_vocab_size, eos_id = create_train_loader(