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

Allow to disable callbacks for some events #156

Closed lassepe closed 11 months ago

lassepe commented 11 months ago

I really like the standardization of logging etc. that FluxTraining comes with. However, for my applications, the logging is much too verbose. Logging the loss after every gradient step significantly slows down my training if the network is fairly small.

In those settings, it would be helpful if the default-callbacks could be told to run at a lower frequency; e.g. every N training examples or even only on at the end of each epoch. Is something like that currently supported without implementing a custom callback?

darsnack commented 11 months ago

Finding things in the docs isn't great right now, but throttle exists already.

lassepe commented 11 months ago

Ah, thank you for the pointer. I'll close this then.