未验证 提交 4ae0c089 编写于 作者: L lvmengsi 提交者: GitHub

[cherry-pick]fix_save (#2673)

上级 fdc0e41a
......@@ -133,7 +133,7 @@ def save_test_image(epoch,
elif cfg.model_net == 'AttGAN' or cfg.model_net == 'STGAN':
for data in zip(A_test_reader()):
real_img, label_org, name = data[0]
attr_names = args.selected_attrs.split(',')
attr_names = cfg.selected_attrs.split(',')
label_trg = copy.deepcopy(label_org)
tensor_img = fluid.LoDTensor()
tensor_label_org = fluid.LoDTensor()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册