I created tf.data.Dataset object with shape ({‘users’:(None,), ‘movies’:(None,)}, (None,)). If I iterate over the dataset object outside model.fit, I see that the shape is same as specified above. But inside model.fit, the input shape changes to {‘users’:(None,1), ‘movies’:(None,1)]. I am using model sub-classing.
Hi @tyagi619
Welcome to the TensorFlow Forum!
The given information is not sufficient to understand the issue. Could you please provide minimal reproducible code to replicate the error? Thank you.