Hi, Thank you for the great work shared. Could you please help to identify correct files to train a model using ImageNet where CIFAR10 related classes (listed in the paper) removed. e.g. to remove n03345487 class how can I update the imagenet_downsampled.py file?