fix cpu (#5212)

PaddleNLP/examples/language_model/transformer-xl/eval.py

@@ -55,6 +55,9 @@ def do_eval(args):
                           (loss, np.exp(loss))
         return logger_info
 
+    if not args.use_gpu:
+        paddle.set_device("cpu")
+
     vocab = get_lm_vocab(args)
     eval_loader = get_lm_data_loader(args, vocab, "valid")
     test_loader = get_lm_data_loader(args, vocab, "test")

PaddleNLP/examples/language_model/transformer-xl/train.py

@@ -37,6 +37,7 @@ def do_train(args):
     else:
         rank = 0
         trainer_count = 1
+        paddle.set_device("cpu")
 
     if trainer_count > 1:
         dist.init_parallel_env()