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