From 887474e9287d4cbe3a54b527dc7b9021b812a81c Mon Sep 17 00:00:00 2001 From: Yang Zhang Date: Tue, 7 Jan 2020 12:11:07 +0800 Subject: [PATCH] Clone for test again after program is popularized --- model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model.py b/model.py index d21e73c..6598a20 100644 --- a/model.py +++ b/model.py @@ -279,7 +279,7 @@ class StaticGraphAdapter(object): return out[:num_output], out[num_output:] def _make_program(self, inputs): - prog = self._orig_prog.clone(for_test=self.mode != 'train') + prog = self._orig_prog.clone() if self.mode == 'train' and self.model._optimizer._learning_rate_map: # HACK workaround learning rate map issue lr_var = self.model._optimizer._learning_rate_map[self._orig_prog] @@ -294,6 +294,8 @@ class StaticGraphAdapter(object): if self.mode == 'train': self._loss_endpoint = fluid.layers.sum(losses) self.model._optimizer.minimize(self._loss_endpoint) + if self.mode != 'train': # clone again to put it in test mode + prog = prog.clone(for_test=True) self._progs[self.mode] = prog self._endpoints[self.mode] = { "output": outputs, -- GitLab