Hi
I was also wondering the same thing.
Callbacks seems like the most convenient place to add and log metrics, but since all the step outputs are saved in memory until the end of the epoch it’s not practical.
is there a way to disable the saving of all step outputs to memory?
Or perhaps a different way to work with callbacks in this manner?
Thanks