deepshah23 commited on
Commit
59cca60
Β·
verified Β·
1 Parent(s): 119db6e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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.2)
81
  ↓ Conv2D(64 β†’ 128) + BatchNorm + ReLU
82
- ↓ MaxPool2d(2x2) + Dropout(0.2)
83
  ↓ Flatten
84
- ↓ Linear(128*7*7 β†’ 128) + BatchNorm + ReLU + Dropout(0.1)
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
  ```