Searched refs:LocalStateDictConfig (Results 1 – 5 of 5) sorted by relevance
9 LocalStateDictConfig,
329 class LocalStateDictConfig(StateDictConfig): class
76 LocalStateDictConfig,701 StateDictType.LOCAL_STATE_DICT: LocalStateDictConfig,
30 .. autoclass:: torch.distributed.fsdp.LocalStateDictConfig
32 LocalStateDictConfig,282 config = LocalStateDictConfig(1016 model, StateDictType.FULL_STATE_DICT, LocalStateDictConfig()