diff options
Diffstat (limited to 'data/aligned_data_loader.py')
| -rw-r--r-- | data/aligned_data_loader.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/data/aligned_data_loader.py b/data/aligned_data_loader.py index 01dbf89..bea3531 100644 --- a/data/aligned_data_loader.py +++ b/data/aligned_data_loader.py @@ -17,9 +17,7 @@ class PairedData(object): return self def __next__(self): - # st() AB, AB_paths = next(self.data_loader_iter) - # st() w_total = AB.size(3) w = int(w_total / 2) h = AB.size(2) @@ -55,7 +53,7 @@ class AlignedDataLoader(BaseDataLoader): batch_size=self.opt.batchSize, shuffle=not self.opt.serial_batches, num_workers=int(self.opt.nThreads)) - + self.dataset = dataset self.paired_data = PairedData(data_loader, opt.fineSize) |