From 316a3b3f501e5365a63f256a23a6ce1cdf6c7971 Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Tue, 9 Apr 2019 11:04:35 +0000 Subject: [PATCH] Don't use persistable for fetched variables --- BERT/run_squad.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/BERT/run_squad.py b/BERT/run_squad.py index dd8caf6..1df3a83 100644 --- 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) -- GitLab