in 2-dl-container/Container-Root/job/resnet/direct_benchmark-inf.py [0:0]
def load_model(file_name): # Load modelbase model = torch.jit.load(file_name) return model