Skip to content

Commit 7b15fd1

Browse files
committed
set_random_seed not support auto_parallel
1 parent 14c651c commit 7b15fd1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

paddleformers/trainer/trainer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,11 @@ def __init__(
369369
self._memory_tracker.start()
370370

371371
# Seed must be set before instantiating the model when using model
372-
set_random_seed(seed_=self.args.seed)
372+
if not self.args.enable_auto_parallel:
373+
set_random_seed(seed_=self.args.seed)
374+
else:
375+
logger.warning("set_seed not support yet in auto_parallel mode")
376+
373377
set_seed(seed=self.args.seed)
374378

375379
self._skip_global_steps = 0 # total skip global steps

0 commit comments

Comments
 (0)