Hugging Face's logo

LWWZH
/
Mini-Vision-V2

Image Classification
PyTorch
minivision_v2
mnist
cnn
computer-vision
mini-vision
mini-vision-series
Model card Files
xet
 Community
Mini-Vision-V2
File size: 1,150 Bytes
f076d9d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
{
    "architectures": ["MiniVisionV2"],
    "model_type": "minivision_v2",
    "framework": "PyTorch",
    "task": "image-classification",
    "num_labels": 10,
    "input_channels": 1,
    "image_size": 28,

    "model_config": {
        "conv1_out_channels": 32,
        "conv2_out_channels": 64,
        "hidden_size": 256,
        "dropout_rate": 0.3,
        "kernel_size": 3
    },

    "training_config": {
        "dataset": "ylecun/mnist",
        "batch_size": 256,
        "learning_rate": 0.01,
        "optimizer": "SGD",
        "momentum": 0.8,
        "scheduler": "StepLR",
        "scheduler_step_size": 3,
        "scheduler_gamma": 0.5,
        "epochs": 50
    },

    "id2label": {
        "0": "0",
        "1": "1",
        "2": "2",
        "3": "3",
        "4": "4",
        "5": "5",
        "6": "6",
        "7": "7",
        "8": "8",
        "9": "9"
    },

    "label2id": {
        "0": "0",
        "1": "1",
        "2": "2",
        "3": "3",
        "4": "4",
        "5": "5",
        "6": "6",
        "7": "7",
        "8": "8",
        "9": "9"
    }
}