Getting this error while trying to train the model using Keras.
raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.InvalidArgumentError: