Searched defs:checkpoint_wrap (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_state_dict.py | 190 self, *fsdp_args, wrap=True, checkpoint_wrap=False, **fsdp_kwargs argument
209 def _get_simple_model(self, *fsdp_args, checkpoint_wrap=False, **fsdp_kwargs): argument
217 self, *fsdp_args, wrap=True, checkpoint_wrap=False, **fsdp_kwargs argument
325 self, state_dict_type, checkpoint_wrap, rank0_only_and_offload argument
|