未验证 提交 b5376779 编写于 作者: K kinghuin 提交者: GitHub

set accelerate_mode default True (#461)

* accelerate_mode default True
上级 ebc18b56
...@@ -30,12 +30,13 @@ if six.PY2: ...@@ -30,12 +30,13 @@ if six.PY2:
else: else:
from inspect import getfullargspec as get_args from inspect import getfullargspec as get_args
import numpy as np import numpy as np
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
from tb_paddle import SummaryWriter from tb_paddle import SummaryWriter
import paddlehub as hub import paddlehub as hub
from paddlehub.common.paddle_helper import dtype_map, clone_program from paddlehub.common.paddle_helper import dtype_map, clone_program
from paddlehub.common.utils import mkdir from paddlehub.common.utils import mkdir, version_compare
from paddlehub.common.dir import tmp_dir from paddlehub.common.dir import tmp_dir
from paddlehub.common.logger import logger from paddlehub.common.logger import logger
from paddlehub.finetune.checkpoint import load_checkpoint, save_checkpoint from paddlehub.finetune.checkpoint import load_checkpoint, save_checkpoint
...@@ -983,7 +984,7 @@ class BaseTask(object): ...@@ -983,7 +984,7 @@ class BaseTask(object):
data, data,
load_best_model=True, load_best_model=True,
return_result=False, return_result=False,
accelerate_mode=False): accelerate_mode=True):
""" """
make prediction for the input data. make prediction for the input data.
...@@ -996,6 +997,11 @@ class BaseTask(object): ...@@ -996,6 +997,11 @@ class BaseTask(object):
Returns: Returns:
RunState: the running result of predict phase RunState: the running result of predict phase
""" """
if not version_compare(paddle.__version__, "1.6.2") and accelerate_mode:
logger.warning(
"Fail to open predict accelerate mode as it does not support paddle < 1.6.2. Please update PaddlePaddle."
)
accelerate_mode = False
self.accelerate_mode = accelerate_mode self.accelerate_mode = accelerate_mode
with self.phase_guard(phase="predict"): with self.phase_guard(phase="predict"):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册