Browse Source

dssm train

丁云鹏 4 months ago
parent
commit
1997ee569e

+ 1 - 1
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IDSSMPredict.java

@@ -1,6 +1,5 @@
 package com.tzld.piaoquan.recommend.model.produce.i2i;
 
-import com.tzld.piaoquan.recommend.model.produce.service.XGBoostService;
 import lombok.extern.slf4j.Slf4j;
 
 /**
@@ -10,6 +9,7 @@ import lombok.extern.slf4j.Slf4j;
 public class I2IDSSMPredict {
 
     public static void main(String[] args) {
+        System.loadLibrary("paddle_inference");
         I2IDSSMService dssm = new I2IDSSMService();
         dssm.predict(args);
     }