When I use pytorch to finetune ResNet, it runs well at the begining, but it stop running after several epoch.
I check nvidia-smi, about half memory is occupied, but GPU is not working, while CPU is almost 100%. It seems like that GPU is waiting for the data from Dataloader which is preprocessed by CPU. I interrupt with CTRL-C, it return some information, can anyong tell me what had happend and how to solve this problem. smaller batch_size, fewer num_workers?  Any advice will be grateful…Thanks