Output_signature of from_generator and Dataset API

Hey! I have a generator that yields TF datasets (of type MapDataset).

I’m trying to feed this generator into Kera’s fit method by:

fit(tf.data.Dataset.from_generator(name_of_gen, output_signature = ?)

How to determine the output signature of the MapDataset object? I’ve tried to get the signature using element_spec but feeding the result into the above doesn’t work…

What’s the correct approach to feeding a generator that outputs Dataset API objects into fit?