mnist-pytorch / README.md
yuweiiizz's picture
feat: initial MNIST model, config, and README
0f61941 verified
---
library_name: pytorch
pipeline_tag: image-classification
tags:
- mnist
- cnn
- pytorch
---
# MNIST CNN (PyTorch)
簡單的 CNN 手寫數字分類器,訓練於 MNIST 資料集。
| Metric | Value |
| ------ | ----- |
| Validation Accuracy | 0.9891 |
| Validation Loss | 0.0345 |
## Usage
```python
import torch
from model import SimpleCNN
model = SimpleCNN()
state = torch.load("pytorch_model.bin")
model.load_state_dict(state)
```