Demeter123 commited on
Commit
a25dbe9
·
verified ·
1 Parent(s): a6f881b

Update README with model details

Browse files
Files changed (1) hide show
  1. README.md +75 -2
README.md CHANGED
@@ -1,2 +1,75 @@
1
- # SimpleCNN for CIFAR-10
2
- Model klasifikasi 10 kelas CIFAR-10 menggunakan CNN sederhana.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - image-classification
5
+ - computer-vision
6
+ - pytorch
7
+ - cnn
8
+ - cifar10
9
+ license: mit
10
+ datasets:
11
+ - cifar10
12
+ model-index:
13
+ - name: SimpleCNN CIFAR-10 Classifier
14
+ results: []
15
+ ---
16
+
17
+ # 🧠 SimpleCNN CIFAR-10 Classifier
18
+
19
+ 📌 A simple Convolutional Neural Network (CNN) model trained on the [CIFAR-10 dataset](https://www.cs.toronto.edu/~kriz/cifar.html), capable of recognizing 10 classes of common objects. The model was trained using PyTorch and is suitable for educational and prototyping purposes.
20
+
21
+ ## 🏷️ Classes
22
+
23
+ - Airplane
24
+ - Automobile
25
+ - Bird
26
+ - Cat
27
+ - Deer
28
+ - Dog
29
+ - Frog
30
+ - Horse
31
+ - Ship
32
+ - Truck
33
+
34
+ ## 🧰 Training Procedure
35
+
36
+ 1. Built a custom CNN model with 3 convolutional layers and 2 fully connected layers.
37
+ 2. Used MaxPooling after each conv layer and dropout for regularization.
38
+ 3. Resized all input images to 32x32 and applied normalization: `(mean=0.5, std=0.5)`.
39
+ 4. Training/validation split:
40
+ - 80% Training
41
+ - 20% Validation
42
+ 5. Training setup:
43
+ - Optimizer: Adam
44
+ - Loss Function: CrossEntropyLoss
45
+ - Batch size: 64
46
+ - Learning rate: 0.001
47
+ - Epochs: 10
48
+ 6. Saved the best-performing model as `pytorch_model.bin`.
49
+
50
+ ## 📊 Performance
51
+
52
+ | Metric | Value |
53
+ |----------------------|-----------|
54
+ | Best Validation Accuracy | 88.76% |
55
+
56
+ ## ⚙️ Framework & Environment
57
+
58
+ - Python: 3.11
59
+ - PyTorch: 2.x (Colab)
60
+ - Torchvision: 0.15.x
61
+ - Platform: Google Colab (GPU enabled)
62
+
63
+ ## 🧪 Hyperparameters
64
+
65
+ | Parameter | Value |
66
+ |-----------------|--------------|
67
+ | Epochs | 10 |
68
+ | Batch Size | 64 |
69
+ | Optimizer | Adam |
70
+ | Learning Rate | 0.001 |
71
+ | Loss Function | CrossEntropy |
72
+ | Image Size | 32x32 |
73
+ | Data Split | 80% Train / 20% Val |
74
+
75
+ ---