From 694c7f0f43ec68a44f6bd835ad0833388f870c17 Mon Sep 17 00:00:00 2001 From: leschultz Date: Thu, 18 Apr 2024 16:21:37 -0500 Subject: [PATCH] Changed example --- examples/materials/combined/fit.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/examples/materials/combined/fit.py b/examples/materials/combined/fit.py index 1f1e158..5c06930 100644 --- 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(