Fix loss calculation

[EKami]

The fix can be found in this timestamp of the course but hasn't been applied to the notebook.

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[jph00]

Thanks!