From a86fa495e19c79e16a2b0184abd6992abaa0ac48 Mon Sep 17 00:00:00 2001 From: junyanz Date: Thu, 20 Apr 2017 03:30:08 -0700 Subject: support different sizes of dataset A/B and update README --- data/aligned_data_loader.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'data/aligned_data_loader.py') 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) -- cgit v1.2.3-70-g09d2