未验证 提交 a4e5b9f3 编写于 作者: T tianxin 提交者: GitHub

Merge pull request #211 from tianxin1860/develop

enable gc for predict_classify
...@@ -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.
先完成此消息的编辑!
想要评论请 注册