Searched defs:_get_base_model (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/fsdp/ | ||
| H A D | test_fsdp_dsd.py | 36 def _get_base_model(self, mlp_dim: int = 2): member in TestFullyShardWithDistributedStateDict |