|
@@ -53,18 +53,12 @@ public class DemoModel implements Model {
|
|
|
String modelDir = PropertiesUtil.getString("model.dir") + "/demo";
|
|
|
CompressUtil.decompressGzFile(in, modelDir);
|
|
|
|
|
|
- String modelFile = modelDir + "/inference.pdmodel";
|
|
|
- String paramFile = modelDir + "/inference.pdiparams";
|
|
|
- File file = new File(modelFile);
|
|
|
- if (file.exists()) {
|
|
|
- log.info("modelFile exists {}", file.getAbsolutePath());
|
|
|
- } else {
|
|
|
- log.info("modelFile not exists {}", file.getAbsolutePath());
|
|
|
- }
|
|
|
+ File modelFile = new File(modelDir + "/inference.pdmodel");
|
|
|
+ File paramFile = new File(modelDir + "/inference.pdiparams");
|
|
|
|
|
|
log.info("start init Config");
|
|
|
Config config = new Config();
|
|
|
- config.setCppModel(modelFile, paramFile);
|
|
|
+ config.setCppModel(modelFile.getAbsolutePath(), paramFile.getAbsolutePath());
|
|
|
config.enableMemoryOptim(true);
|
|
|
config.enableProfile();
|
|
|
config.enableMKLDNN();
|
|
@@ -74,7 +68,7 @@ public class DemoModel implements Model {
|
|
|
|
|
|
log.info("start init Predictor");
|
|
|
Predictor predictor = Predictor.createPaddlePredictor(config);
|
|
|
- Predictor temp = predictor;
|
|
|
+ Predictor temp = sourcePredictor;
|
|
|
this.sourcePredictor = predictor;
|
|
|
temp.destroyNativePredictor();
|
|
|
return true;
|