Downloads · 30 days
0
train_infer
Tasfiya025/train_infer
train_infer is a machine learning model from Tasfiya025. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Model: ResNet18 (fine-tuned) Framework: PyTorch Intended Use: - Small projects - Robotics object classification - Educational ML experiments
Access
Public
Updated Dec 11, 2025
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
.py3.5 KB · 63%
At a glance
- Access
- Public
- Created
- Dec 11, 2025
- Updated
- Dec 11, 2025
- SHA
- 65bcc7ee
Model card
From the Hugging Face model README
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.
- Created
- Dec 11, 2025
- Updated
- Dec 11, 2025