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 ca79444 commit 13fd655Copy full SHA for 13fd655
paddlenlp/trainer/trainer.py
@@ -626,7 +626,7 @@ def train(
626
# The resume_from_checkpoint could be None in some machine node.
627
# Here we reset None to temp directory.
628
if args.world_size > 1:
629
- is_resume_from_checkpoint = paddle.to_tensor([resume_from_checkpoint is not None])
+ is_resume_from_checkpoint = paddle.to_tensor([resume_from_checkpoint is not None], dtype="int32")
630
paddle.distributed.all_reduce(is_resume_from_checkpoint)
631
is_resume_from_checkpoint = is_resume_from_checkpoint.item()
632
if is_resume_from_checkpoint > 0 and is_resume_from_checkpoint < paddle.distributed.get_world_size():
0 commit comments