diff --git a/tools/export_serving_model.py b/tools/export_serving_model.py index 8c2310eb75706298a3b5742552c88b58082d98de..368ee157599eb08087f4f344b8cb09aa84ab0138 100644 --- a/tools/export_serving_model.py +++ b/tools/export_serving_model.py @@ -31,7 +31,7 @@ from ppdet.utils.check import check_config, check_version, enable_static_mode import ppdet.utils.checkpoint as checkpoint import yaml import logging -from export_model import parse_reader, dump_infer_config, prune_feed_vars +from ppdet.utils.export_utils import dump_infer_config, prune_feed_vars FORMAT = '%(asctime)s-%(levelname)s: %(message)s' logging.basicConfig(level=logging.INFO, format=FORMAT) logger = logging.getLogger(__name__)