Compute_total_loss_test ex 6 fails

[saifkhanengr]

What about this point:

@paulinpaloalto

Make sure you specify the positional arguments to the loss function in the correct order.

First, you have to pass labels and then logits.