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)