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

Fix save (#2670)

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