    def train(self):
        """
        For training model
        """
        pass
For training model
tests/unittests/mock_operators/pytorch_image_classification_operator/pytorch_image_classification_operator.py