Thanks for the link! I learned something new from that.
I accidentally managed to make it work by reconnecting the kernel of the Jupyter notebook. Seems like it was the kernel who got confused on what to do.
However, I encountered another issue. When I set up the num_workers to more than 0 for the loader variable, the training get frozen and doesn’t want to proceed. I have waited for quite a while already and nithong seemingly happens. Would you have some idea what it could be?