updated model.py

model.py

@@ -20,7 +20,7 @@ from modchallenge.interface.base_model import ModularMultiplicationModel
 MAX_DIGITS = 10      # decimal digits per input slot (covers p up to ~4 billion = tier 4)
 MAX_OUT_DIGITS = 10  # decimal digits in the answer (answer < p <= tier-4 max)
 INPUT_SIZE = 3 * MAX_DIGITS          # 30
-HIDDEN_SIZE = 512
+HIDDEN_SIZE = 1024
 OUTPUT_SIZE = MAX_OUT_DIGITS * 10    # 100
 
 
@@ -39,6 +39,10 @@ class VanillaMLP(nn.Module):
         self.net = nn.Sequential(
             nn.Linear(input_size, hidden_size),
             nn.ReLU(),
+            nn.Linear(hidden_size, hidden_size),
+            nn.ReLU(),
+            nn.Linear(hidden_size, hidden_size),
+            nn.ReLU(),
             nn.Linear(hidden_size, output_size),
         )