Inquiry about the compatibility of the code with the ImageNet dataset

[a2108870]

Hi，

I hope this message finds you well. I wanted to ask if your code is compatible with ImageNet.