n_query=15, **few_shot_params) novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples) elif params.test_dataset == "EuroSAT": print("Loading EuroSAT") datamgr = EuroSAT_few_shot.SetDataManager2(image_size, n_eposide=iter_num, n_query=15, **few_shot_params) novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples) elif params.test_dataset == "CropDisease": print("Loading CropDisease") datamgr = CropDisease_few_shot.SetDataManager2(image_size, n_eposide=iter_num, n_query=15, **few_shot_params) novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples) elif params.test_dataset == "ChestX": print("Loading ChestX") datamgr = Chest_few_shot.SetDataManager2(image_size, n_eposide=iter_num, n_query=15, **few_shot_params) novel_loader = datamgr.get_data_loader( num_aug=params.gen_examples) ### what if aug is true??? ## uncomment code below to see if code is same across loaders #########################################################################
outfile = os.path.join( checkpoint_dir.replace("checkpoints","features"), params.dataset + "_" + str(params.save_iter)+ ".hdf5") else: outfile = os.path.join( checkpoint_dir.replace("checkpoints","features"), params.dataset + ".hdf5") if params.dataset in ["ISIC"]: datamgr = ISIC_few_shot.SimpleDataManager(image_size, batch_size = 64) data_loader = datamgr.get_data_loader(aug = False ) elif params.dataset in ["EuroSAT"]: datamgr = EuroSAT_few_shot.SimpleDataManager(image_size, batch_size = 64) data_loader = datamgr.get_data_loader(aug = False ) elif params.dataset in ["CropDisease"]: datamgr = CropDisease_few_shot.SimpleDataManager(image_size, batch_size = 64) data_loader = datamgr.get_data_loader(aug = False ) elif params.dataset in ["ChestX"]: datamgr = Chest_few_shot.SimpleDataManager(image_size, batch_size = 64) data_loader = datamgr.get_data_loader(aug = False ) model = model_dict[params.model]() model = model.cuda() tmp = torch.load(modelfile) state = tmp['state'] state_keys = list(state.keys()) for i, key in enumerate(state_keys):