We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6ff5eab commit 08a2275Copy full SHA for 08a2275
examples/machine_translation/transformer/static/train.py
@@ -99,6 +99,7 @@ def do_train(args):
99
places = paddle.static.cuda_places()
100
elif args.device == "xpu":
101
places = paddle.static.xpu_places()
102
+ paddle.set_device("xpu")
103
else:
104
places = paddle.static.cpu_places()
105
paddle.set_device("cpu")
0 commit comments