from main.experiments import torch, parser, RTPT
from main.experiments import run_model_imagefolder

torch.set_num_threads(6)
args = parser.parse_args()

dir_name = '/workspace/datasets/imagenet21k/imagenet21k_resized/imagenet21k_val'
save_dir = 'imagenet21k_val'

run_model_imagefolder(args, dir_name, save_dir)
