Method train

examples/cnn_ms/pkg_model_code/model.py:213–220

Set the model in evaluation mode. Args: mode(bool): when mode is True, this model will enter training mode

(self, mode=True)

        raise NotImplementedError

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

        Args:
            mode(bool): when mode is True, this model will enter training mode
        """
        self.training = mode
        autograd.training = mode

    def eval(self):
        """Sets the model in evaluation mode.