Callback list seems to be a good option. I will try it out.



 markdaoust:

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


If your subclassed model (where I am overriding train_step()) contains two or more models and if you are passing ModelCheckpoint callback while calling .fit() on the subclassed model the callback would get confused.