We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
torch.distributed.checkpoint.state_dict.set_model_state_dict
load_checkpoint_in_model