diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-20 06:35:55 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-20 06:35:55 -0700 |
| commit | 63761c5d8abea169d8a03a73c0bae99aaf036b6a (patch) | |
| tree | 77b54e0e5dd2c551257b0129aa78980e8d23232a | |
| parent | bce1bbd63c469b5fdec88abad614060c66cd7132 (diff) | |
fix a bug in unaligned_data_loader
| -rw-r--r-- | README.md | 2 | ||||
| -rw-r--r-- | data/unaligned_data_loader.py | 4 |
2 files changed, 5 insertions, 1 deletions
@@ -62,7 +62,7 @@ bash ./datasets/download_cyclegan_dataset.sh maps ``` - Train a model: ```bash -# ./scripts/train_cyclegan.sh +#!./scripts/train_cyclegan.sh python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan ``` - To view training results and loss plots, run `python -m visdom.server` and click the URL http://localhost:8097. To see more intermediate results, check out `./checkpoints/maps_cyclegan/web/index.html` diff --git a/data/unaligned_data_loader.py b/data/unaligned_data_loader.py index 4f82dbe..4a06510 100644 --- a/data/unaligned_data_loader.py +++ b/data/unaligned_data_loader.py @@ -13,6 +13,8 @@ class PairedData(object): self.stop_B = False def __iter__(self): + self.stop_A = False + self.stop_B = False self.data_loader_A_iter = iter(self.data_loader_A) self.data_loader_B_iter = iter(self.data_loader_B) return self @@ -37,6 +39,8 @@ class PairedData(object): B, B_paths = next(self.data_loader_B_iter) if self.stop_A and self.stop_B: + self.stop_A = False + self.stop_B = False raise StopIteration() else: return {'A': A, 'A_paths': A_paths, |