I am getting an error in the train_model() function. Not sure where the error is originating from. My data_generator() function seems to be fine and passes all the unit tests.
I am getting an error in the train_model() function. Not sure where the error is originating from. My data_generator() function seems to be fine and passes all the unit tests.
Thanks @arvyzukai - My error is resolved now.
And yes it was the axis issue