Ok I changed num_workers to 0 from 2.
It did seem to run past 1 step as it then gave a variable name error inside the if statement (which was my fault). However, that 1 step took like 40+ minutes. I will add a timer now to get an exact time but this script may take days to run it seems. Am I right in thinking that the solution maybe be a Learning Rate Scheduler? Although would this have any effect after just one step?
I also can’t run on the terminal as I am coding from a Mac and can’t use the cuda, hence why I am doing it through the normal Kaggle notebooks, which is also hosting the data too.