tensorplay.distributed.checkpoint.set_state_dict

tensorplay.distributed.checkpoint.set_state_dict(model: Any, optimizers: Any, *, model_state_dict: dict[str, Any], optim_state_dict: dict[str, Any], options: StateDictOptions | None = None) → Any[source]