Epochs keep running , though epochs=10

TensorFlow General Discussion
1

Epoch 1/10
136/3967 [>…] - ETA: 2:10:02 - loss: 1.6152 - accuracy: 0.4414

InvalidArgumentError Traceback (most recent call last)
in <cell line: 2>()
1 epochs=10
----> 2 history = model.fit(
3 train_ds,
4 validation_data=val_ds,
5 epochs=epochs

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)
50 try:
51 ctx.ensure_initialized()
—> 52 tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
53 inputs, attrs, num_outputs)
54 except core._NotOkStatusException as e:

InvalidArgumentError: Graph execution error:

3

Hi @Spring_Ireland, Could you please provide the standalone code to reproduce the issue. Thank You.