The code in the linked notebook uses F.nll_loss with F.log_softmax as the last activation in the model, which is correct.
Your model seems to use F.softmax, so could you change it to F.log_softmax and rerun the code, please?