Pythoncodist/Projects
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update neural network code

Browse Source
This commit is contained in:
Donald Calloway 2025-09-24 09:43:30 -07:00
parent 06d6cbbbac
commit 28d2083a03
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
neural_network_example.py
View File

@ -19,7 +19,7 @@ class TinyModel(nn.Module):
x = self.layer2(x) x = self.layer2(x)
return x return x
model = TinyModel() model = TinyModel()
loss_fn = nn.MSELoss() loss_fn = nn.MSELoss()
optimizer = optim.SGD(model.parameters(), lr=0.01) optimizer = optim.SGD(model.parameters(), lr=0.01)