diff --git a/ERNIE/predict_classifier.py b/ERNIE/predict_classifier.py index ecda8a3b77cee416755b7d32d6a4e1486fdc6a63..e6115dabe31e56ef698c3632bc5fd0eea865db17 100644 --- a/ERNIE/predict_classifier.py +++ b/ERNIE/predict_classifier.py @@ -22,6 +22,12 @@ import time 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