File size: 441 Bytes
215d2b8 0f61941 215d2b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
---
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)
```
|