I get this error when I train on bottleneck.
IndexError: index 160 is out of bounds for axis 0 with size 160
This is the code that the error says it has the proble
