Skip to content

Instantly share code, notes, and snippets.

@AbdelmajidB
Created June 21, 2019 10:34
Show Gist options
  • Save AbdelmajidB/67c6683f2eba156e265fd4cd6121d275 to your computer and use it in GitHub Desktop.
Save AbdelmajidB/67c6683f2eba156e265fd4cd6121d275 to your computer and use it in GitHub Desktop.
Exception in thread "main" java.lang.IllegalStateException: Input and label arrays do not have same shape: [24, 3, 96, 96] vs. [24, 1, 96, 96]
at org.nd4j.base.Preconditions.throwStateEx(Preconditions.java:641)
at org.nd4j.base.Preconditions.checkState(Preconditions.java:340)
at org.deeplearning4j.nn.layers.convolution.CnnLossLayer.backpropGradient(CnnLossLayer.java:80)
at org.deeplearning4j.nn.graph.vertex.impl.LayerVertex.doBackward(LayerVertex.java:149)
at org.deeplearning4j.nn.graph.ComputationGraph.calcBackpropGradients(ComputationGraph.java:2663)
at org.deeplearning4j.nn.graph.ComputationGraph.computeGradientAndScore(ComputationGraph.java:1378)
at org.deeplearning4j.nn.graph.ComputationGraph.computeGradientAndScore(ComputationGraph.java:1338)
at org.deeplearning4j.optimize.solvers.BaseOptimizer.gradientAndScore(BaseOptimizer.java:160)
at org.deeplearning4j.optimize.solvers.StochasticGradientDescent.optimize(StochasticGradientDescent.java:63)
at org.deeplearning4j.optimize.Solver.optimize(Solver.java:52)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment