diff --git a/BERT/run_squad.py b/BERT/run_squad.py
index dd8caf6dbafd1b6a9dbcb1987e636559e9456b7e..1df3a83cb778fc618657bce40cb96b40984375a1 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)