Message: axis 2 is out of bounds for array of dimension 2

[imran-suny]

[jacobgil]

Flatten-514 won't work as a target layer. It has to be the 2D output of a CNN. So Block8-511 for example or earlier, before the adaptive pooling that turns it into 1d