Method train_one_batch

examples/cnn_ms/pkg_model_code/model.py:202–211

Defines the computation performed in every training iteration Should be overridden by all subclasses. Args: *input: the arguments of train_one_batch **kwargs: the keyword arguments of train_one_batch

(self, *input, **kwargs)

        raise NotImplementedError

    def train_one_batch(self, *input, **kwargs):
        """Defines the computation performed in every training iteration

        Should be overridden by all subclasses.

        Args:
            *input: the arguments of train_one_batch
            **kwargs: the keyword arguments of train_one_batch
        """
        raise NotImplementedError

    def train(self, mode=True):
        """Set the model in evaluation mode.