train_infer / README.md
Tasfiya025's picture
Create README.md
65bcc7e verified
# Model Card – Simple ResNet18 Image Classifier
**Model:** ResNet18 (fine-tuned)
**Framework:** PyTorch
**Intended Use:**
- Small projects
- Robotics object classification
- Educational ML experiments
**Training Data:**
User-provided images placed in:
```
data/train/<class>
data/val/<class>
```
**Evaluation Metric:**
- Validation Accuracy per epoch.
**Limitations:**
- Not suitable for medical/safety-critical tasks.
- Accuracy depends fully on dataset quality.
**License:** Free for personal and commercial use.