Hi,
I see also in model declaration:
layers.OutputLayer(256, (2, 2))

I may explain the (batch x 2 x 2).
I got first question layers print from: “print(model)” and it actually is not very precise, as you point out.
Is there any way to print model with more detail?
Thanks,