Don't use persistable for fetched variables

316a3b3f · Yibing Liu · 6b8f27b2 · 316a3b3f
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

BERT/run_squad.py BERT/run_squad.py +2 -1

未找到文件。
--- a/BERT/run_squad.py
+++ b/BERT/run_squad.py
@@ -300,7 +300,8 @@ def train(args):
                    bert_config=bert_config,
                    is_training=False)

-                fluid.memory_optimize(test_prog, skip_opt_set=[unique_ids.name, start_logits.name, end_logits.name, num_seqs.name])
+                fluid.memory_optimize(test_prog, skip_opt_set=[unique_ids.name,
+                    start_logits.name, end_logits.name, num_seqs.name])

        test_prog = test_prog.clone(for_test=True)