tensorplay.distributed.checkpoint.load_sharded_optimizer_state_dict

tensorplay.distributed.checkpoint.load_sharded_optimizer_state_dict(model_state_dict: dict[str, Any], optimizer_key: str, storage_reader: Any, planner: LoadPlanner | None = None) → dict[str, Any][source]