Downloads · 30 days
0
mnist-modal
ivanleomk/mnist-modal
mnist-modal is a machine learning model from ivanleomk. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This is a PyTorch model trained on the MNIST dataset using an NVIDIA A100 GPU on Modal.
Access
Public
Updated Apr 24, 2026
Repo size
9.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth4.8 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Apr 24, 2026
- Updated
- Apr 24, 2026
- SHA
- c08b0125
Model card
From the Hugging Face model README
MNIST Model Trained on Modal
This is a PyTorch model trained on the MNIST dataset using an NVIDIA A100 GPU on Modal.
Model Details
- Architecture: Simple CNN (Conv2D -> Conv2D -> MaxPool -> Dropout -> FC -> Dropout -> FC)
- Dataset: MNIST
- Final Test Accuracy: 99.02%
Links
- W&B Run: View training on Weights & Biases
- Training Script: The script used to train this model is available in the repository as
train_mnist.py.
Training Configuration
- Batch Size: 64
- Learning Rate: 1.0 (Adadelta)
- Epochs: 5
Training Metrics
| Epoch | Train Loss | Test Loss | Accuracy |
|---|---|---|---|
| 1 | 0.2036 | 0.0441 | 98.40% |
| 2 | 0.0828 | 0.0376 | 98.84% |
| 3 | 0.0622 | 0.0487 | 98.38% |
| 4 | 0.0528 | 0.0301 | 99.05% |
| 5 | 0.0466 | 0.0314 | 99.02% |
- License
- mit
- Created
- Apr 24, 2026
- Updated
- Apr 24, 2026