Merge pull request #741 from panyx0718/use_program_cache

Add program caching
PaddlePaddle · Mar 19, 2018 · c7a1c88 · c7a1c88
2 parents d224397 + 82ed838
commit c7a1c88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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))