Searched refs:_model_step_with_zero_grad (Results 1 – 1 of 1) sorted by relevance
4185 def _model_step_with_zero_grad(self, model): member in DistributedTest._DistTestBase4256 self._model_step_with_zero_grad(model_base)4257 self._model_step_with_zero_grad(model_DDP)6176 self._model_step_with_zero_grad(model_DDP)