Issue in train step of custom keras model when passing data from a generator

I found a very similar issue to the one I am having on stack overflow here, but I’m posting here because no one figured it out. That post pretty much explains the issue I am having, but I am using an ImageDataGenerator instead. I cant seem to be able to get it to work within the train_step of a custom keras model. Any help is appreciated

submitted by /u/proxygonn
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *