diff --git a/PaddleCV/gan/trainer/CycleGAN.py b/PaddleCV/gan/trainer/CycleGAN.py index a8b9561e5cbaf28fce532bf8aac67a2c6ad6c166..751695f991698f8fdd4543246c38e13a80c36f7f 100644 --- a/PaddleCV/gan/trainer/CycleGAN.py +++ b/PaddleCV/gan/trainer/CycleGAN.py @@ -96,11 +96,11 @@ class GTrainer(): learning_rate=fluid.layers.piecewise_decay( boundaries=[99 * step_per_epoch] + [ x * step_per_epoch - for x in xrange(100, cfg.epoch - 1) + for x in range(100, cfg.epoch - 1) ], values=[lr] + [ lr * (1.0 - (x - 99.0) / 101.0) - for x in xrange(100, cfg.epoch) + for x in range(100, cfg.epoch) ]), beta1=0.5, beta2=0.999, @@ -136,11 +136,11 @@ class DATrainer(): learning_rate=fluid.layers.piecewise_decay( boundaries=[99 * step_per_epoch] + [ x * step_per_epoch - for x in xrange(100, cfg.epoch - 1) + for x in range(100, cfg.epoch - 1) ], values=[lr] + [ lr * (1.0 - (x - 99.0) / 101.0) - for x in xrange(100, cfg.epoch) + for x in range(100, cfg.epoch) ]), beta1=0.5, beta2=0.999, @@ -175,11 +175,11 @@ class DBTrainer(): learning_rate=fluid.layers.piecewise_decay( boundaries=[99 * step_per_epoch] + [ x * step_per_epoch - for x in xrange(100, cfg.epoch - 1) + for x in range(100, cfg.epoch - 1) ], values=[lr] + [ lr * (1.0 - (x - 99.0) / 101.0) - for x in xrange(100, cfg.epoch) + for x in range(100, cfg.epoch) ]), beta1=0.5, beta2=0.999, diff --git a/PaddleCV/gan/util/utility.py b/PaddleCV/gan/util/utility.py index d14878a8f6ac12c90a96e95aacbc3eb4626c1a8f..aacfb1e6c332174a7d7d8d180a425729bd325fdc 100644 --- a/PaddleCV/gan/util/utility.py +++ b/PaddleCV/gan/util/utility.py @@ -149,7 +149,8 @@ def save_test_image(epoch, for j in range(len(label_org)): label_trg_tmp[j][i] = 1.0 - label_trg_tmp[j][i] - label_trg_ = map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp) + label_trg_ = list( + map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp)) for j in range(len(label_org)): label_trg_[j][i] = label_trg_[j][i] * 2.0