Searched refs:_get_simple_nested_model (Results 1 – 2 of 2) sorted by relevance
189 def _get_simple_nested_model( member in TestFSDPStateDict265 self._get_simple_nested_model(*fsdp_args, wrap=wrap, **fsdp_kwargs),266 self._get_simple_nested_model(*fsdp_args, wrap=wrap, **fsdp_kwargs),345 partial(self._get_simple_nested_model),554 self._get_simple_nested_model,711 model = self._get_simple_nested_model(mixed_precision=mixed_precision)1186 model = self._get_simple_nested_model()
246 def _get_simple_nested_model( member in TestFSDPMixedPrecision407 self._get_simple_nested_model(884 else self._get_simple_nested_model