def METHOD_NAME(self):
    if self.model_type == "LeNet":
        return LeNet()
    elif self.model_type == "resnet56":
        return resnet56(10, pretrained=False, path=None)
    elif self.model_type == "resnet20":
        return resnet20(10)
    else:
        raise Exception(f"do not support this model: {self.model_type}")