def build_dataset_test(self):
"""
Create testing dataset
"""
self.dataset_test = dataset.SURREAL(train=False)
self.dataloader_test = torch.utils.data.DataLoader(self.dataset_test, batch_size=5,
shuffle=False, num_workers=int(self.opt.workers),
drop_last=True)
self.len_dataset_test = len(self.dataset_test)
def build_dataset_train(self):
"""
Create training dataset
"""
self.dataset_train = dataset.SURREAL(train=True, regular_sampling=True)
self.dataloader_train = torch.utils.data.DataLoader(self.dataset_train, batch_size=self.opt.batch_size,
shuffle=True, num_workers=int(self.opt.workers),
drop_last=True)
self.len_dataset = len(self.dataset_train)