Skip to content

Commit 6c22d40

Browse files
fix UT
1 parent 8d71b11 commit 6c22d40

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

source/tests/pd/model/test_deeppot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def setUp(self) -> None:
4242
trainer = get_trainer(deepcopy(self.config))
4343
trainer.run()
4444

45-
with paddle.device("cpu"):
45+
with paddle.device.device_guard("cpu"):
4646
input_dict, label_dict, _ = trainer.get_data(is_train=False)
4747
trainer.wrapper(**input_dict, label=label_dict, cur_lr=1.0)
4848
self.model = "model.pd"

0 commit comments

Comments
 (0)