RuntimeError: index 1002 is out of bounds for dimension 1 with size 1000

[18222312562]

When I tried to run this code on ImageNet1k dataset I met this problem. However, I didn't modify the code except some parameters such as batch size, model, workers.
The image including the error information is attached. Is there anything wrong? Or is there anything I could do to fix it?
Thank you for offering the code, looking forward to your reply!

[yuweihao]

Hi @18222312562 , you may check the MLP classification head and the number of folders in your ImageNet.