From 637827a5bc80d6e0a17466e44b087f91601539cb Mon Sep 17 00:00:00 2001 From: Jeff Wang Date: Fri, 8 Jun 2018 15:05:25 -0700 Subject: [PATCH] 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 --- python/paddle/fluid/inferencer.py | 2 ++ python/paddle/fluid/trainer.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/inferencer.py b/python/paddle/fluid/inferencer.py index 9f242cf29a5..6baac009057 100644 --- a/python/paddle/fluid/inferencer.py +++ b/python/paddle/fluid/inferencer.py @@ -56,6 +56,8 @@ class Inferencer(object): else: self.exe = executor.Executor(self.place) + self.inference_program = self.inference_program.clone(for_test=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 diff --git a/python/paddle/fluid/trainer.py b/python/paddle/fluid/trainer.py index cdacb419863..ac313b237ee 100644 --- a/python/paddle/fluid/trainer.py +++ b/python/paddle/fluid/trainer.py @@ -115,9 +115,9 @@ class Trainer(object): program_func_outs = train_func() self.train_func_outputs = program_func_outs if isinstance( 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] optimizer = optimizer_func() -- GitLab