Translations:PyTorch/311/en
Jump to navigation
Jump to search
if global_rank == 0:
torch.save(ddp_model.state_dict(), "./checkpoint_path")