diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:39:31 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-20 03:39:31 -0700 |
| commit | d4206aad119326c57fefeb97176f8fbda6cd8d1e (patch) | |
| tree | ab5aed69e069c1d446918f13c28049d156c6be90 /data/aligned_data_loader.py | |
| parent | 03d01ea7723015b29aac078daa2d2797e042923a (diff) | |
| parent | 443bc13554769d6a18eefdbac779cf385fb6dbb3 (diff) | |
merged conflicts
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) |