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) | |
``` | |