save model

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/WarmUpService.java

@@ -17,9 +17,7 @@ import javax.annotation.PostConstruct;
 public class WarmUpService {
     @PostConstruct
     public void warmup() {
-        // ModelManager.getInstance().registerModel(ModelEnum.VIDEO_DSSM);
-
         ModelManager.getInstance().registerModel(ModelEnum.DEMO);
-        ModelManager.getInstance().registerModel(ModelEnum.DNN);
+        ModelManager.getInstance().registerModel(ModelEnum.VIDEO_DSSM);
     }
 }

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DSSMModel.java

@@ -26,9 +26,16 @@ public class DSSMModel implements Model {
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
 
+        for (int i = 0; i < predictor.getInputNum(); i++) {
+            log.info("predictor2 inName{}={}", i, predictor.getInputNameById(i));
+        }
+
+        for (int i = 0; i < predictor.getOutputNum(); i++) {
+            log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
+        }
         // 2 设置输入
-        inHandle.reshape(4, new int[]{1, 3, 224, 224});
-        float[] inData = new float[1 * 3 * 224 * 224];
+        inHandle.reshape(2, new int[]{1, 47});
+        float[] inData = new float[1 * 47];
         inHandle.copyFromCpu(inData);
 
         // 3 预测

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/ModelEnum.java

@@ -3,7 +3,7 @@ package com.tzld.piaoquan.recommend.model.service.model;
 import org.apache.commons.lang3.StringUtils;
 
 public enum ModelEnum {
-    VIDEO_DSSM("videoDssm", "dyp/dssm.tar.gz", DSSMModel.class),
+    VIDEO_DSSM("videoDssm", "dyp/dssm_demo.tar.gz", DSSMModel.class),
     DEMO("demo", "zhangbo/model_paddle_demo.tar.gz", DemoModel.class),
     DNN("dnn", "dyp/dnn.tar.gz", DNNModel.class),
     NULL("null", "null", null);