Add ability to record and log arbitrary learner values

FluxML / FluxTraining.jl

A flexible neural net training library inspired by fast.ai

https://fluxml.ai/FluxTraining.jl

MIT License

117 stars 25 forks source link

Add ability to record and log arbitrary learner values #136

Closed darsnack closed 1 year ago

darsnack commented 1 year ago

This adds two callbacks, Traces and LogTraces, for recording and logging values from the learner, respectively. This essentially makes it possible to record step state that we don't have default recorders for.

lorenzoh commented 1 year ago

LGTM! 👍

Feel free to merge and tag a new version when you want!