I'm facing the below issue in my code implementation, can you anybody help me?

Epoch 1/5

InvalidArgumentError Traceback (most recent call last)
in <cell line: 1>()
----> 1 history = model.fit(train_generator, epochs = 5, validation_data = valid_generator, verbose = 1)

1 frames
/usr/local/lib/python3.10/dist-packages/tensorflow/python/eager/execute.py in quick_execute(op_name, num_outputs, inputs, attrs, ctx, name)
51 try:
52 ctx.ensure_initialized()
—> 53 tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
54 inputs, attrs, num_outputs)
55 except core._NotOkStatusException as e:

InvalidArgumentError: Graph execution error:

Hi @Pushpavathi_Kothapal, If possible could you please provide the standalone code to reproduce the issue.Thank You.