resnet50 / README.md
kevin-nd's picture
Upload README.md with huggingface_hub
2a91d61 verified
---
datasets:
- imagenet
language: en
library_name: pytorch
license: mit
pipeline_tag: image-classification
tags:
- image-classification
- pytorch
- lightning
- imagenet
---
# ResNet50 trained on ImageNet-1k
This model was trained using the [cv_101](https://github.com/K-bNd/cv_101) deep learning framework.
- **Task:** image-classification
- **Architecture:** ResNet50
- **Dataset:** ImageNet-1k
- **Framework:** PyTorch + PyTorch Lightning
## Test Results
| Metric | Value |
|--------|-------|
| Accuracy | 0.7476 |
| Top-5 Accuracy | 0.9197 |
| Loss | 2.1267 |