소스 검색

dssm train

丁云鹏 6 달 전
부모
커밋
1997ee569e
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IDSSMPredict.java

+ 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);
     }