Error with implementing compute_total_loss

Here is a checklist of the most common errors on that function.

But generally speaking, it’s more useful to show us your actual output, as opposed to just saying “I’m having a problem”. That would give us more information to go on. Please note that no-one else can actually look at your notebooks: that information is private to you.