I need to get the shape of the input X data when training begins. But, I do not see the data passed into:
def on_train_begin(self, logs=None):

This is a bummer because it’s really easy in pytorch. Is there a workaround to do this?