Each time I am trying to call training function, It’s showing some broadcasting error. I checked the data_generator function also final_model layers but didn’t find any error.
Please suggest that hidden mistake.
https://colab.research.google.com/drive/10seKwt3mqe7Tn3eRnnhJNld5ErFHPwD_