Also have you implemted def metrics(self) or called reset_states() if you have used a custom model with a custom train loop?

  
      

      TensorFlow
  

  
    

Customize what happens in Model.fit  |  TensorFlow Core 2