Searched refs:_init_transformer_model (Results 1 – 1 of 1) sorted by relevance
315 _ModelClass.TRANSFORMER: self._init_transformer_model,353 def _init_transformer_model( member in TestFSDPOptimState