model.py

class TrainModel(object):
    def input(self):
        pass

    def net(self):
        pass

    def net(self):
        pass

    def loss(self):
        pass

    def optimizer(self):
        pass


class InferModel(object):
    def input(self):
        pass

    def net(self):
        pass

    def net(self):
        pass

    def loss(self):
        pass

    def optimizer(self):
        pass