diff options
Diffstat (limited to '')
| -rw-r--r-- | projects/autograd.md | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/projects/autograd.md b/projects/autograd.md index f1dce97..59f70d2 100644 --- a/projects/autograd.md +++ b/projects/autograd.md @@ -25,10 +25,10 @@ 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() |