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 724b524 commit ed42f16Copy full SHA for ed42f16
paddlenlp/trainer/trainer.py
@@ -631,7 +631,7 @@ def train(
631
# The resume_from_checkpoint could be None in some machine node.
632
# Here we reset None to temp directory.
633
if args.world_size > 1:
634
- 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")
635
paddle.distributed.all_reduce(is_resume_from_checkpoint)
636
is_resume_from_checkpoint = is_resume_from_checkpoint.item()
637
if is_resume_from_checkpoint > 0 and is_resume_from_checkpoint < paddle.distributed.get_world_size():
0 commit comments