I am trying to build up some custom generator and put its returns into keras layers for bath-size training. It says I had some attribute issues, yet I have no idea of how to