from yqn_pytorch_framework.model.base_model import BaseModel
from ${module_name}.model.pytorch_${module_name}_layer import *


class ${class_name}Model(BaseModel):

    def __init__(self, config):
        super(${class_name}Model, self).__init__(config)
        self.config = config

    def forward(self, inputs):
        # TODO fill ${module_name} model forward
        pass

    def get_default_input(self):
        # TODO fill ${module_name} model get_default_input
        pass
