File size: 396 Bytes
e290690 ad6bd6f e290690 ad6bd6f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
---
license: apache-2.0
datasets:
- dataautogpt3/Dalle3
- scrapegraphai/AQL-v1-QA
language:
- en
metrics:
- accuracy
base_model:
- microsoft/resnet-50
new_version: microsoft/resnet-50
pipeline_tag: image-classification
---
```python
def load_model(model_path, num_classes):
model = create_model(num_classes)
model.load_state_dict(torch.load(model_path))
model.eval()
return model |