It Worked 
You rock big time!
If I understand correctly the change was that you replaced loss_func=CrossEntropyLosst() with loss_func=CrossEntropyLossFlat()
Can you please explain how this made a difference and more importantly how you found that was the problem and how you found the solution?
You are amazing. Thank you so much!
P.s.it’s very strange that the problem was with the learner code as it’s copy pasted from Jeremy’s lesson…