Spaces:

abdurafay19
/

Digit-Classifier

Sleeping

Abdul Rafay commited on Mar 10

Commit

d4b6872

1 Parent(s): a661514

updated model

Files changed (1) hide show

model.py CHANGED Viewed

@@ -17,9 +17,11 @@ class Model(nn.Module):
         self.fc_layers = nn.Sequential(
             nn.Flatten(),
-            nn.Linear(128 * 7 * 7, 64),
             nn.ReLU(),
-            nn.Linear(64, 32),
             nn.ReLU(),
             nn.Linear(32, 10)
         )

         self.fc_layers = nn.Sequential(
             nn.Flatten(),
+            nn.Linear(128 * 7 * 7, 84),
             nn.ReLU(),
+            nn.Linear(84, 56),
+            nn.ReLU(),
+            nn.Linear(56, 32),
             nn.ReLU(),
             nn.Linear(32, 10)
         )