Method training_step

python/oneflow/framework/model.py:206–209

Operates on a single batch of data from the training set and return loss.

(self, *args, **kwargs)

        raise NotImplementedError

    def training_step(self, *args, **kwargs):
        """Operates on a single batch of data from the training set and return loss.
        """
        raise NotImplementedError()

    def validation_step(self, *args, **kwargs):
        """Operates on a single batch of data from the validation set.