tensorplay.distributed.checkpoint.get_state_dict

tensorplay.distributed.checkpoint.get_state_dict(model: Any, optimizers: Any, *, submodules: set[Any] | None = None, options: StateDictOptions | None = None) → tuple[dict[str, Any], dict[str, Any]][source]