def load_datasets(raw_data, split_idx, args):
data_splits = data_utils.read_splits('%s/split_%d.txt' %
(args.data, split_idx))
dataset_loaders = {}
if not args.test_mode:
dataset_loaders['train'] = get_loader(raw_data,
data_splits['train'],
args,
shuffle=True)
dataset_loaders['valid'] = get_loader(raw_data,
data_splits['valid'],
args,
shuffle=False)
dataset_loaders['test'] = get_loader(raw_data,
data_splits['test'],
args,
shuffle=False)
return dataset_loaders
def get_test_loader(raw_data, split_idx, args):
data_splits = data_utils.read_splits('%s/split_%d.txt' %
(args.data, split_idx))
test_loader = get_loader(raw_data,
data_splits['test'],
args,
shuffle=False,
num_workers=0,
batch_size=1)
return test_loader