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