ashikshafi0:

tf.data.Dataset.from_generator()


Looping in @Andrew_Audibert 
I also checked for similar Issues here (keyword: tf.data.Dataset.from_generator -  Issues · tensorflow/tensorflow · GitHub 13) - in case someone has encountered a similar issue.