Merge pull request #211 from tianxin1860/develop

enable gc for predict_classify
PaddlePaddle · Jul 18, 2019 · a4e5b9f · a4e5b9f
2 parents 85cf2ee + 8dc2d4b
commit a4e5b9f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ERNIE/predict_classifier.py b/ERNIE/predict_classifier.py
@@ -22,6 +22,12 @@
 import argparse
 import numpy as np
 import multiprocessing
+
+# NOTE(paddle-dev): All of these flags should be
+# set before `import paddle`. Otherwise, it would
+# not take any effect.
+os.environ['FLAGS_eager_delete_tensor_gb'] = '0'  # enable gc
+
 import paddle.fluid as fluid
 
 from reader.task_reader import ClassifyReader