Closely related to:
https://discuss.tensorflow.org/t/model-checkpointing-best-practices-when-using-train-step
When I am using this the callback is unable to keep track of the metric values correctly:

image2336×460 58.6 KB

I notice something similar in the SimSiam example I linked in my previous post as well. Am I missing out on?