Open yangjun023 opened 6 years ago
Can you please show full error message, and what you're feeding there?
1.the log as follows:
Exception in thread "main" java.lang.IllegalStateException: Mis matched lengths: [640000] != [5000]
at org.nd4j.linalg.util.LinAlgExceptions.assertSameLength(LinAlgExceptions.java:40)
at org.nd4j.linalg.api.ops.BaseTransformOp.<init>(BaseTransformOp.java:47)
at org.nd4j.linalg.api.ops.impl.transforms.Set.<init>(Set.java:24)
at org.nd4j.linalg.api.ndarray.BaseNDArray.assign(BaseNDArray.java:1267)
at org.nd4j.linalg.api.ndarray.BaseNDArray.reshape(BaseNDArray.java:3753)
at org.deeplearning4j.nn.graph.vertex.impl.ReshapeVertex.doForward(ReshapeVertex.java:77)
at org.deeplearning4j.nn.graph.ComputationGraph.feedForward(ComputationGraph.java:1464)
at org.deeplearning4j.nn.graph.ComputationGraph.silentOutput(ComputationGraph.java:1555)
at org.deeplearning4j.nn.graph.ComputationGraph.output(ComputationGraph.java:1544)
at org.deeplearning4j.nn.graph.ComputationGraph.outputSingle(ComputationGraph.java:1579)
the parameters are: batch=128, doclength=50
outputSingle,this will return the p of labels
1.the model use ReshapePreProcessor