Skip to content

Instantly share code, notes, and snippets.

@AbdelmajidB

AbdelmajidB/error

Last active May 14, 2019 14:17
Show Gist options
  • Save AbdelmajidB/1fd463351400eacfbec9ff901ecc6eca to your computer and use it in GitHub Desktop.
Save AbdelmajidB/1fd463351400eacfbec9ff901ecc6eca to your computer and use it in GitHub Desktop.
Download ZIP
Raw
error
Exception in thread "main" org.deeplearning4j.exception.DL4JInvalidInputException: Got rank 4 array as input to Convolution3DLayer (layer name = conv1-1, layer index = 1) with shape [1, 1, 256, 256]. Expected rank 5 array with shape [minibatchSize, numChannels, inputHeight, inputWidth, inputDepth]. (layer name: conv1-1, layer index: 1, layer type: Convolution3DLayer)
at org.deeplearning4j.nn.layers.convolution.Convolution3DLayer.preOutput(Convolution3DLayer.java:189)
at org.deeplearning4j.nn.layers.convolution.ConvolutionLayer.activate(ConvolutionLayer.java:437)
at org.deeplearning4j.nn.graph.vertex.impl.LayerVertex.doForward(LayerVertex.java:111)
at org.deeplearning4j.nn.graph.ComputationGraph.ffToLayerActivationsInWS(ComputationGraph.java:2116)
at org.deeplearning4j.nn.graph.ComputationGraph.computeGradientAndScore(ComputationGraph.java:1369)
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment