I have a quite large neural network and it is hard to create minimal reproducible example with the problem. I have been using Keras backend so far, but decided
Keras