diff options
Diffstat (limited to 'projects/autograd.md')
| -rw-r--r-- | projects/autograd.md | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/projects/autograd.md b/projects/autograd.md index fe076eb..0a9997f 100644 --- a/projects/autograd.md +++ b/projects/autograd.md @@ -25,15 +25,15 @@ X = [ [ 3.0, 3.0, 3.0 ] ] -y = [ 1.0, -1.0, 1.0, -1.0 ] + y = [ 1.0, -1.0, 1.0, -1.0 ] n = MLP(3, [ 4, 4, 1 ]) -for i in range(400): + for i in range(400): pred = [ n(x) for x in X ] loss = mse(y, pred) loss.zero_grad() loss.backward() n.optimise(0.01) -print(pred) -``` + print(pred) + ``` |