tensorplay.distributed.checkpoint.set_optimizer_state_dict

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