diff --git a/lite/demo/python/mobilenetv1_full_api.py b/lite/demo/python/mobilenetv1_full_api.py index c3a6bd077be5978f1ecaf9b040b119e50117d62b..7d713dfeea2735c008c0870c6758bba3236ad506 100644 --- a/lite/demo/python/mobilenetv1_full_api.py +++ b/lite/demo/python/mobilenetv1_full_api.py @@ -20,9 +20,6 @@ from __future__ import division from __future__ import print_function import argparse -import sys -sys.path.append('../../python/lib') - from paddlelite.lite import * # Command arguments @@ -42,8 +39,8 @@ def RunModel(args): config.set_param_file(args.param_file) else: config.set_model_dir(args.model_dir) - # For x86, you can set places = [Place(TargetType.X86, PrecisionType.FP32)] - places = [Place(TargetType.ARM, PrecisionType.FP32)] + # For arm platform (armlinux), you can set places = [Place(TargetType.ARM, PrecisionType.FP32)] + places = [Place(TargetType.X86, PrecisionType.FP32)] config.set_valid_places(places) # 2. Create paddle predictor diff --git a/lite/demo/python/mobilenetv1_light_api.py b/lite/demo/python/mobilenetv1_light_api.py index 5847c7819366b654dd9d5b5cbe2108b54da7b04c..a3e2db6c0df382b96cddfd5578e295fc7d47a357 100644 --- a/lite/demo/python/mobilenetv1_light_api.py +++ b/lite/demo/python/mobilenetv1_light_api.py @@ -20,8 +20,6 @@ from __future__ import division from __future__ import print_function import argparse -import sys -sys.path.append('../../python/lib') from paddlelite.lite import * @@ -33,7 +31,7 @@ parser.add_argument( def RunModel(args): # 1. Set config information config = MobileConfig() - config.set_model_dir(args.model_dir) + config.set_model_from_file(args.model_dir) # 2. Create paddle predictor predictor = create_paddle_predictor(config)