Parcourir la source

model service

丁云鹏 il y a 5 mois
Parent
commit
dc5be22080

+ 7 - 0
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DemoModel.java

@@ -8,6 +8,7 @@ import com.tzld.piaoquan.recommend.model.util.JSONUtils;
 import com.tzld.piaoquan.recommend.model.util.PropertiesUtil;
 import lombok.extern.slf4j.Slf4j;
 
+import java.io.File;
 import java.io.InputStream;
 
 @Slf4j
@@ -54,6 +55,12 @@ public class DemoModel implements Model {
 
             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());
+            }
 
             log.info("start init Config");
             Config config = new Config();