提交 8dc2d4b8 编写于 作者: T tianxin

enable gc for predict_classify

上级 2f2fe7af
...@@ -22,6 +22,12 @@ import time ...@@ -22,6 +22,12 @@ import time
import argparse import argparse
import numpy as np import numpy as np
import multiprocessing 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 import paddle.fluid as fluid
from reader.task_reader import ClassifyReader from reader.task_reader import ClassifyReader
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册