A newer version of this model is available: microsoft/resnet-50
def load_model(model_path, num_classes):
model = create_model(num_classes)
model.load_state_dict(torch.load(model_path))
model.eval()
return model
Model tree for anurag2506/coat_classification
Base model
microsoft/resnet-50