def _init_train_loader(self):
dataset = ds.NumpyMRFDataset(self.params['train_database_dir'])
loader = data.DataLoader(dataset,
batch_size=self.params['batch_size_training'],
shuffle=True,
num_workers=0)
return loader
def _init_valid_loader(self):
dataset = ds.NumpyMRFDataset(self.params['valid_database_dir'])
loader = data.DataLoader(dataset,
batch_size=self.params['batch_size_testing'],
shuffle=False,
num_workers=0)
return loader