def __init__(self, device, meta_val_iter, meta_val_way, meta_val_shot, meta_val_query, ex): self.train_loader = get_dataloader(split='train', aug=True, shuffle=True) sample_info = [ meta_val_iter, meta_val_way, meta_val_shot, meta_val_query ] self.val_loader = get_dataloader(split='val', aug=False, sample=sample_info) self.device = device self.num_classes = ex.current_run.config['model']['num_classes']
def get_loaders(self, used_set, target_data_path, target_split_dir): # First, get loaders loaders_dic = {} loader_info = {'aug': False, 'shuffle': False, 'out_name': False} if target_data_path is not None: # This mean we are in the cross-domain scenario loader_info.update({'path': target_data_path, 'split_dir': target_split_dir}) train_loader = get_dataloader('support', **loader_info) loaders_dic['support'] = train_loader test_loader = get_dataloader('query', **loader_info) loaders_dic.update({'query': test_loader}) return loaders_dic