Training Loop (Again!) in UNQ_C8

Hi @Rama_Mahajanam

I private messaged you suggestions for your code.

For future readers - do not import additional libraries when you are not asked to do that. And in general, please check these points.

Good luck!