From 82ed838007d58b73a512d42bcad09c878f0a69ac Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Mon, 19 Mar 2018 04:29:02 -0700 Subject: [PATCH] Add program caching Reduce step time from 0.2 to 0.15 --- fluid/neural_machine_translation/transformer/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fluid/neural_machine_translation/transformer/train.py b/fluid/neural_machine_translation/transformer/train.py index 19835c48..0494792c 100644 --- a/fluid/neural_machine_translation/transformer/train.py +++ b/fluid/neural_machine_translation/transformer/train.py @@ -140,7 +140,8 @@ def main(): lr_scheduler.update_learning_rate(data_input) outs = exe.run(fluid.framework.default_main_program(), feed=data_input, - fetch_list=[cost]) + fetch_list=[cost], + use_program_cache=True) cost_val = np.array(outs[0]) print("pass_id = " + str(pass_id) + " batch = " + str(batch_id) + " cost = " + str(cost_val)) -- GitLab