Vishal_R:

all. In jupyter notebook, it shows that cell is being executed but no output is shown. So I am forced to run with num_workers=0. Anything above 0 breaks for some reason.


try typing
watch nvidia-smi

in your shell at the same time while training is happening it will show the real memory usage of your model and utilization.