yuweiiizz/mnist-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)