From 28d2083a03f01c44639cd2f5f520756dd7c33785 Mon Sep 17 00:00:00 2001 From: Donald Calloway Date: Wed, 24 Sep 2025 09:43:30 -0700 Subject: [PATCH] Update neural network code --- neural_network_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_network_example.py b/neural_network_example.py index 2cbb9f5..21e255c 100644 --- a/neural_network_example.py +++ b/neural_network_example.py @@ -19,7 +19,7 @@ class TinyModel(nn.Module): x = self.layer2(x) return x - model = TinyModel() +model = TinyModel() loss_fn = nn.MSELoss() optimizer = optim.SGD(model.parameters(), lr=0.01)