fix exampmle.src.md

2020-03-23 15:16:54 +08:00 · 2020-03-23 15:16:54 +08:00 · 41020767af
parent fb4188fc83
commit 41020767af
1 changed files with 1 additions and 1 deletions
--- a/notebook/example.src.md
+++ b/notebook/example.src.md
@ -59,7 +59,7 @@ optim = nn.SGD (model.parameters(), learning_rate)

 for i,(x,y) in enumerate(get_data(n)):
    pred_y = model(x)
-    loss = ((pred_y - y)**2)
+    loss = jt.sqr(pred_y - y)
    loss_mean = loss.mean()
    optim.step (loss_mean)
    print(f"step {i}, loss = {loss_mean.data.sum()}")