|
@@ -34,8 +34,8 @@ public class DNNModel implements Model {
|
|
|
log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
|
|
|
}
|
|
|
|
|
|
- inHandle.reshape(2, new int[]{1, 1});
|
|
|
- float[] inData = new float[1 * 1];
|
|
|
+ inHandle.reshape(2, new int[]{1, 158});
|
|
|
+ float[] inData = new float[1 * 158];
|
|
|
inHandle.copyFromCpu(inData);
|
|
|
predictor.run();
|
|
|
|