Sayak_Paul:

How do I set up a ModelCheckpoint callback for this one?


Could you clarify the question? Why doesn’t  it work to just pass the callback to .fit like you normally would?



 Bhack:

But also in your case if you manually populate the CallbackList


Yes. If you’re writing your own training loop, you need to drive the callbacks section using callback list here:

  
      

      TensorFlow
  

  
    

tf.keras.callbacks.Callback  |  TensorFlow Core v2.5.0 3

  Abstract base class used to build new callbacks.


  

  
    
    
  

  


But if you’er using .train_step and .fit, all the callbacks should be driven as normal, no?