Update README.md
Browse files
README.md
CHANGED
@@ -77,11 +77,11 @@ These transformations simulate handwritten noise and variation in real student s
|
|
77 |
Input: (1, 28, 28)
|
78 |
β Conv2D(1 β 32) + BatchNorm + ReLU
|
79 |
β Conv2D(32 β 64) + BatchNorm + ReLU
|
80 |
-
β MaxPool2d(2x2) + Dropout(0.
|
81 |
β Conv2D(64 β 128) + BatchNorm + ReLU
|
82 |
-
β MaxPool2d(2x2) + Dropout(0.
|
83 |
β Flatten
|
84 |
-
β Linear(128*7*7 β 128) + BatchNorm + ReLU + Dropout(0.
|
85 |
β Linear(128 β 11)
|
86 |
β Output: class logits (digits 0β9, blank = 10)
|
87 |
```
|
|
|
77 |
Input: (1, 28, 28)
|
78 |
β Conv2D(1 β 32) + BatchNorm + ReLU
|
79 |
β Conv2D(32 β 64) + BatchNorm + ReLU
|
80 |
+
β MaxPool2d(2x2) + Dropout(0.1)
|
81 |
β Conv2D(64 β 128) + BatchNorm + ReLU
|
82 |
+
β MaxPool2d(2x2) + Dropout(0.1)
|
83 |
β Flatten
|
84 |
+
β Linear(128*7*7 β 128) + BatchNorm + ReLU + Dropout(0.2)
|
85 |
β Linear(128 β 11)
|
86 |
β Output: class logits (digits 0β9, blank = 10)
|
87 |
```
|