Invalid Argument Error

I have a problem with the code, unfortunately I don’t quite know what I should do next. This is what the error message looks like


and here is the code
image

Hi @Blazej

Welcome to the TensorFlow Forum!

What is the dataset shape and the conv_base model description? You can try by removing the steps_per_epoch args from the model.fit() while model training.

Please provide the reproducible code to replicate the error if the issue still persists. Thank you.