Can you provide the code to retrain model?

[TrieuDinhHo]

TypeError: in user code:

File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py", line 1972, in test_function  *
    return step_function(self, iterator)
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py", line 1956, in step_function  **
    outputs = model.distribute_strategy.run(run_step, args=(data,))
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py", line 1944, in run_step  **
    outputs = model.test_step(data)
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py", line 1852, in test_step
    self.compute_loss(x, y, y_pred, sample_weight)
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/training.py", line 1140, in compute_loss
    y, y_pred, sample_weight, regularization_losses=self.losses
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/base_layer.py", line 1429, in losses
    loss_tensor = regularizer()
File "/usr/local/lib/python3.10/dist-packages/keras/src/engine/base_layer.py", line 1515, in _tag_callable
    loss = tf.convert_to_tensor(loss, dtype=backend.floatx())

TypeError: can't convert Operation 'StatefulPartitionedCall' to Tensor

I met this error when trying to retrain the model