diff --git a/PaddleCV/PaddleGAN/infer.py b/PaddleCV/PaddleGAN/infer.py index 17bb550e186e68c5cb6810334d76efea725f4636..9ebe94433f18b7246a731222e1751714d29df1e7 100644 --- a/PaddleCV/PaddleGAN/infer.py +++ b/PaddleCV/PaddleGAN/infer.py @@ -177,8 +177,8 @@ def infer(args): label_trg_ = list( map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp)) if args.model_net == 'AttGAN': - for j in range(len(label_org)): - label_trg_[j][i] = label_trg_[j][i] * 2.0 + for k in range(len(label_org)): + label_trg_[k][i] = label_trg_[k][i] * 2.0 tensor_label_org_.set(label_org_, place) tensor_label_trg.set(label_trg, place) tensor_label_trg_.set(label_trg_, place) diff --git a/PaddleCV/PaddleGAN/util/utility.py b/PaddleCV/PaddleGAN/util/utility.py index 8a48ab284d634f01a778fd580a737438e57238c4..2f55a922216d1530d23b59c07d712bf9e0ccfe82 100644 --- a/PaddleCV/PaddleGAN/util/utility.py +++ b/PaddleCV/PaddleGAN/util/utility.py @@ -160,8 +160,8 @@ def save_test_image(epoch, map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp)) if cfg.model_net == 'AttGAN': - for j in range(len(label_org)): - label_trg_[j][i] = label_trg_[j][i] * 2.0 + for k in range(len(label_org)): + label_trg_[k][i] = label_trg_[k][i] * 2.0 tensor_label_org_.set(label_org_, place) tensor_label_trg.set(label_trg, place) tensor_label_trg_.set(label_trg_, place)