Skip to content

Commit 156b899

Browse files
authored
Merge pull request PaddlePaddle#1845 from liuwei1031/develop
set persistable for fetched variables temporarily to pass CE, test=develop
2 parents 8c3d380 + 2cf6350 commit 156b899

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

fluid/PaddleCV/gan/cycle_gan/trainer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ def __init__(self, input_A, input_B):
1313
self.program = fluid.default_main_program().clone()
1414
with fluid.program_guard(self.program):
1515
self.fake_B = build_generator_resnet_9blocks(input_A, name="g_A")
16+
#FIXME set persistable explicitly to pass CE
17+
self.fake_B.persistable = True
1618
self.fake_A = build_generator_resnet_9blocks(input_B, name="g_B")
1719
self.cyc_A = build_generator_resnet_9blocks(self.fake_B, "g_B")
1820
self.cyc_B = build_generator_resnet_9blocks(self.fake_A, "g_A")
@@ -58,6 +60,8 @@ def __init__(self, input_A, input_B):
5860
with fluid.program_guard(self.program):
5961
self.fake_B = build_generator_resnet_9blocks(input_A, name="g_A")
6062
self.fake_A = build_generator_resnet_9blocks(input_B, name="g_B")
63+
#FIXME set persistable explicitly to pass CE
64+
self.fake_A.persistable = True
6165
self.cyc_A = build_generator_resnet_9blocks(self.fake_B, "g_B")
6266
self.cyc_B = build_generator_resnet_9blocks(self.fake_A, "g_A")
6367
self.infer_program = self.program.clone()

0 commit comments

Comments
 (0)