DIVA-DIA / DIVA-DAF

Repository for the deep-learning framework DIVA-DAF which is build with historical document image analysis in mind.
https://diva-dia.github.io/DIVA-DAF/
MIT License
18 stars 2 forks source link

Timer callback #156

Closed lvoegtlin closed 6 months ago

lvoegtlin commented 11 months ago

Is your feature request related to a problem? Please describe. Extend the Timer callback to be able to log the after every training epoch the time and also log the end running time of the fit and the evaluate stages.

Describe the solution you'd like Inherit from the original callback and just add the logging functionality on the main thread to make it log on the default loggers.

Checklist

lvoegtlin commented 6 months ago

The original callback if for fixing the run time of the experiment and not track the elapsed time. We have to create our own custom callback