diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-20 03:30:08 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-20 03:30:08 -0700 |
| commit | a86fa495e19c79e16a2b0184abd6992abaa0ac48 (patch) | |
| tree | ae817e3948779e7d372e695d519cb9b12fa6b575 /data/aligned_data_loader.py | |
| parent | baa154d611b8d8d4104b40a767d643cac3f52dec (diff) | |
support different sizes of dataset A/B and update README
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) |