def create_datasets(args):
train_data = SliceData(args.train_path, args.acceleration_factor,
args.dataset_type)
dev_data = SliceData(args.validation_path, args.acceleration_factor,
args.dataset_type)
return dev_data, train_data
def create_datasets(args):
train_data = SliceData(args.dataset_path, args.acceleration_factor,
'train')
dev_data = SliceData(args.dataset_path, args.acceleration_factor,
'validation')
return dev_data, train_data
def create_data_loaders(args):
data = SliceData(args.data_path, args.acceleration_factor, 'validation')
data_loader = DataLoader(
dataset=data,
batch_size=args.batch_size,
num_workers=1,
pin_memory=True,
)
return data_loader