From 2710b7d4549a126ad57d975e2ad5a2c2b260d118 Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Tue, 28 Jun 2022 10:28:06 +0800 Subject: [PATCH] [BugFix] Remove is_init_py of RandomGenerator (#2658) --- paddlenlp/trainer/trainer_base.py | 1 - 1 file changed, 1 deletion(-) diff --git a/paddlenlp/trainer/trainer_base.py b/paddlenlp/trainer/trainer_base.py index 8fa78fb8e1a6..84aae2211d89 100644 --- a/paddlenlp/trainer/trainer_base.py +++ b/paddlenlp/trainer/trainer_base.py @@ -877,7 +877,6 @@ def _load_rng_state(self, checkpoint): core = paddle.framework.core if core.is_compiled_with_cuda(): for i in range(core.get_cuda_device_count()): - core.default_cuda_generator(i)._is_init_py = True core.default_cuda_generator(i).manual_seed( checkpoint_rng_state["cuda"][i])