Update 07_1_nonlinear_regression.py

kmiddleton · kmiddleton · commit df1fd1db109a · 2024-09-29T10:32:52.000-05:00
diff --git a/07_1_nonlinear_regression.py b/07_1_nonlinear_regression.py
@@ -64,7 +64,7 @@ def forward(self, x):
 optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)  
 
 # 3) Training loop
-num_epochs = 10000
+num_epochs = 20000
 loss_values = []
 
 for epoch in range(num_epochs):
