Method __call__

examples/cnn_ms/pkg_model_code/model.py:241–245

(self, *input, **kwargs)

        return self.__class__.__name__

    def __call__(self, *input, **kwargs):
        if self.training:
            return self.train_one_batch(*input, **kwargs)
        else:
            return self.forward(*input, **kwargs)

    def save_states(self, fpath, aux_states={}):
        """Save states.