model.py

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Function
import models


def build_model(cfg, input_size):    
    return eval(f"models.{cfg['Model']['Name']}(cfg, input_size)")

if __name__== "__main__":
    import yaml
    config_file = open("config.yml","r")
    cfg = yaml.load(config_file)
    print(cfg['Model']['Name'])