|
@@ -34,8 +34,8 @@ public class DSSMModel implements Model {
|
|
|
log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
|
|
|
}
|
|
|
// 2 设置输入
|
|
|
- inHandle.reshape(2, new int[]{1, 47});
|
|
|
- float[] inData = new float[1 * 47];
|
|
|
+ inHandle.reshape(2, new int[]{1, 157});
|
|
|
+ float[] inData = new float[1 * 157];
|
|
|
inHandle.copyFromCpu(inData);
|
|
|
|
|
|
// 3 预测
|