Changed example

leschultz · Apr 18, 2024 · 694c7f0 · 694c7f0
1 parent 06a4990
commit 694c7f0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/examples/materials/combined/fit.py b/examples/materials/combined/fit.py
@@ -26,7 +26,11 @@ def main():
         data[k]['scaler'] = StandardScaler()
         data[k]['loss'] = nn.L1Loss()
 
-    model = models.MultiNet(tasks=tasks, input_arch={500: 1})
+    model = models.MultiNet(
+                            tasks=tasks,
+                            input_arch={500: 1},
+                            mid_arch={1024: 1, 32: 1, 16: 1},
+                            )
     optimizer = optim.Adam
 
     out = utils.train(