未验证 提交 637827a5 编写于 作者: J Jeff Wang 提交者: GitHub

Use for_test=True in the Fluid Trainer to clone the test program (#11323)

* Use for_test=True in the Fluid Trainer to clone the test program

* fix typo

* Should do the samething to the inferencer
上级 52e2eb65
...@@ -56,6 +56,8 @@ class Inferencer(object): ...@@ -56,6 +56,8 @@ class Inferencer(object):
else: else:
self.exe = executor.Executor(self.place) self.exe = executor.Executor(self.place)
self.inference_program = self.inference_program.clone(for_test=True)
def infer(self, inputs, return_numpy=True): def infer(self, inputs, return_numpy=True):
""" """
:param inputs: a map of {"input_name": input_var} that will be feed into the inference program :param inputs: a map of {"input_name": input_var} that will be feed into the inference program
......
...@@ -115,9 +115,9 @@ class Trainer(object): ...@@ -115,9 +115,9 @@ class Trainer(object):
program_func_outs = train_func() program_func_outs = train_func()
self.train_func_outputs = program_func_outs if isinstance( self.train_func_outputs = program_func_outs if isinstance(
program_func_outs, list) else [program_func_outs] program_func_outs, list) else [program_func_outs]
self.test_program = self.train_program.clone() self.test_program = self.train_program.clone(for_test=True)
# The fisrt element of program_func_outs is loss. # The first element of program_func_outs is loss.
loss = self.train_func_outputs[0] loss = self.train_func_outputs[0]
optimizer = optimizer_func() optimizer = optimizer_func()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册