def load_model()

in 2-dl-container/Container-Root/job/bert/direct_benchmark-inf.py [0:0]


def load_model(file_name):
    # Load modelbase
    model = torch.jit.load(file_name)

    return model