metadata
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
import torch
from model import SimpleCNN
model = SimpleCNN()
state = torch.load("pytorch_model.bin")
model.load_state_dict(state)