Method step

deepspeed/runtime/pipe/engine.py:1311–1313

Disabled for pipeline parallel training. See ``train_batch()``.

(self, *args, **kwargs)

        raise PipelineError("Only train_batch() is accessible in pipeline mode.")

    def step(self, *args, **kwargs):
        """Disabled for pipeline parallel training. See ``train_batch()``. """
        raise PipelineError("Only train_batch() is accessible in pipeline mode.")

    def module_state_dict(self, exclude_frozen_parameters=False):
        """Override hack to save a pipe model and return the directory path of the save.