Parameters: 970 (<1000) Test accuracy: 92.35% Epochs trained: 45 (early stopping after 5 epochs without improvement)
Source du modèle
Description de la source
This model was trained on RX 6600.
Sources
1 sourceVérifié 28 sept.
Artefacts du modèle
1 artefactExtraits de sources
2 extraits--- language: en license: cc0-1.0 tags: - mnist - tiny-model - early-stopping pipeline_tag: image-classification --- # Tiny MNIST Classifier - **Parameters**: 970 (<1000) - **Test accuracy**: 92.35% - **Epochs trained**: 45 (early stopping after 5 epochs without improvement) This model was trained on RX 6600. ## Full results | Metric | Value | |---------------------------|-----------------| | Total parameters | 970 | | Best validation loss | 0.2463 | | Final test accuracy | 92.35% | | Early stopping patience | 5 | | Training epochs | 45 | ## Model architecture AvgPool(4x4) → Linear(49→16) → ReLU → Dropout(0.2) → Linear(16→10) ## How to use ```python import torch from train import TinyMNISTModel model = TinyMNISTModel() model.load_state_dict(torch.load("mnist_1k_best.pth")) model.eval() ```
| Metric | Value |
|---|
| Total parameters | 970 |
| Best validation loss | 0.2463 |
| Final test accuracy | 92.35% |
| Early stopping patience | 5 |
| Training epochs | 45 |
AvgPool(4x4) → Linear(49→16) → ReLU → Dropout(0.2) → Linear(16→10)
import torch
from train import TinyMNISTModel
model = TinyMNISTModel()
model.load_state_dict(torch.load("mnist_1k_best.pth"))
model.eval()
Source context: 0 downloads · 0 likes · Pipeline image-classification · Repo sapbot/tiny-mnist-classifier