Update README.md
Browse files
README.md
CHANGED
|
@@ -5,6 +5,7 @@ tags:
|
|
| 5 |
- mnist
|
| 6 |
- tiny-model
|
| 7 |
- early-stopping
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# Tiny MNIST Classifier
|
|
@@ -38,4 +39,4 @@ from train import TinyMNISTModel
|
|
| 38 |
model = TinyMNISTModel()
|
| 39 |
model.load_state_dict(torch.load("mnist_1k_best.pth"))
|
| 40 |
model.eval()
|
| 41 |
-
```
|
|
|
|
| 5 |
- mnist
|
| 6 |
- tiny-model
|
| 7 |
- early-stopping
|
| 8 |
+
pipeline_tag: image-classification
|
| 9 |
---
|
| 10 |
|
| 11 |
# Tiny MNIST Classifier
|
|
|
|
| 39 |
model = TinyMNISTModel()
|
| 40 |
model.load_state_dict(torch.load("mnist_1k_best.pth"))
|
| 41 |
model.eval()
|
| 42 |
+
```
|