def get_model(channels_last=False, ipex_optimize=False, compile=False): model = torchvision.models.resnet50() if channels_last: model= model.to(memory_format=torch.channels_last) model = model.eval() if ipex_optimize: model = ipex.optimize(model, dtype=torch.bfloat16) if compile: model = torch.compile(model) return model