提交 00b9e0a9 编写于 作者: T TeslaZhao

update PrecisionType.Float32 import path

上级 da09478a
...@@ -20,6 +20,7 @@ import google.protobuf.text_format ...@@ -20,6 +20,7 @@ import google.protobuf.text_format
import numpy as np import numpy as np
import argparse import argparse
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.inference as inference
from .proto import general_model_config_pb2 as m_config from .proto import general_model_config_pb2 as m_config
from paddle.fluid.core import PaddleTensor from paddle.fluid.core import PaddleTensor
from paddle.fluid.core import AnalysisConfig from paddle.fluid.core import AnalysisConfig
...@@ -125,11 +126,10 @@ class LocalPredictor(object): ...@@ -125,11 +126,10 @@ class LocalPredictor(object):
if use_lite: if use_lite:
config.enable_lite_engine( config.enable_lite_engine(
precision_mode = PrecisionType.Float32, precision_mode=inference.PrecisionType.Float32,
zero_copy = True, zero_copy=True,
passes_filter = [], passes_filter=[],
ops_filter = [] ops_filter=[])
)
if use_xpu: if use_xpu:
config.enable_xpu(100 * 1024 * 1024) config.enable_xpu(100 * 1024 * 1024)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册