Skip to content

Commit 494a8a1

Browse files
authored
[LLM] fix normal sl (#10609)
1 parent afa102d commit 494a8a1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/trainer/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2722,7 +2722,7 @@ def _save_checkpoint(self, model, metrics=None):
27222722
] = fleet.meta_parallel.get_rng_state_tracker().get_states_tracker()
27232723

27242724
if self.args.save_rng_states:
2725-
rng_state_file = f"rng_state_{dist.get_rank()}.pth"
2725+
rng_state_file = os.path.join(output_dir, f"rng_state_{dist.get_rank()}.pth")
27262726
os.makedirs(output_dir, exist_ok=True)
27272727
paddle.save(rng_states, rng_state_file)
27282728

0 commit comments

Comments
 (0)