Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
3D CNN throwing ND4JIllegalStateException #6366
Here is a standalone test class that throws:
It uses a dummy custom RecordReader that returns 10 3D records filled with 0 and with label 0.
So, looking at this: there's a couple of issues - one on your side, and one on ours.
First, 3D CNN data/activations have 5 dimensions, not 3. See this for more details:
For single examples, your leading dimension should be 1
Second: RecordReaderMultiDataSetIterator hadn't been set up for 5D data via NDArrayWritables.