yuweiiizz commited on
Commit
215d2b8
·
verified ·
1 Parent(s): e86acf8

feat: initial MNIST model, config, and README

Browse files
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ pipeline_tag: image-classification
4
+ tags:
5
+ - mnist
6
+ - cnn
7
+ - pytorch
8
+ ---
9
+
10
+ # MNIST CNN (PyTorch)
11
+
12
+ 簡單的 CNN 手寫數字分類器,訓練於 MNIST 資料集。
13
+
14
+ | Metric | Value |
15
+ | ------ | ----- |
16
+ | Validation Accuracy | 0.9898 |
17
+ | Validation Loss | 0.0320 |
18
+
19
+ ## Usage
20
+ ```python
21
+ import torch
22
+ from model import SimpleCNN
23
+ model = SimpleCNN()
24
+ state = torch.load("pytorch_model.bin")
25
+ model.load_state_dict(state)
26
+ ```
config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input_shape": [
3
+ 1,
4
+ 28,
5
+ 28
6
+ ],
7
+ "framework": "pytorch",
8
+ "architecture": "SimpleCNN",
9
+ "epochs": 3
10
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:011564823e370e36bab7aa8f7d8f7f8fcba008ec10a4e5e94e93cca19d57640a
3
+ size 4802952
tb_logs/events.out.tfevents.1754325079.329cdbd66de4.600.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40985d51bb68d6af47d1cf8ad9f8d9c3bb5f92b2227c47a850ea1816f1f60fa1
3
+ size 65624