提交 2cf63500 编写于 作者: L liuwei1031

set persistable for fetched variables temporarily to pass CE, test=develop

上级 f001d4b8
......@@ -13,6 +13,8 @@ class GATrainer():
self.program = fluid.default_main_program().clone()
with fluid.program_guard(self.program):
self.fake_B = build_generator_resnet_9blocks(input_A, name="g_A")
#FIXME set persistable explicitly to pass CE
self.fake_B.persistable = True
self.fake_A = build_generator_resnet_9blocks(input_B, name="g_B")
self.cyc_A = build_generator_resnet_9blocks(self.fake_B, "g_B")
self.cyc_B = build_generator_resnet_9blocks(self.fake_A, "g_A")
......@@ -58,6 +60,8 @@ class GBTrainer():
with fluid.program_guard(self.program):
self.fake_B = build_generator_resnet_9blocks(input_A, name="g_A")
self.fake_A = build_generator_resnet_9blocks(input_B, name="g_B")
#FIXME set persistable explicitly to pass CE
self.fake_A.persistable = True
self.cyc_A = build_generator_resnet_9blocks(self.fake_B, "g_B")
self.cyc_B = build_generator_resnet_9blocks(self.fake_A, "g_A")
self.infer_program = self.program.clone()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册