search

sdatkinson / neural-amp-modeler

Neural network emulator for guitar amplifiers.
MIT License
1.72k stars 130 forks source link

[FEATURE] Add "Stop training if ESR equal to" option in GUI "Advanced options" #393

Closed 2-dor closed 3 months ago

2-dor commented 3 months ago

Hey Steve,

Been training models in batches; some turn out pretty good in fewer epochs, others need more.

It would be of great help if the GUI trainer could have an option that stops training the current model (and skips to the next one if more are lined up) if it's already at a "good-enough" quality level for the user running the training.

Something like a checkbox for "Stop training if ESR is equal to" and a field value where the user can input their desired ESR value like "0.015" etc.

Thanks for taking the time to read this 😊

sdatkinson commented 3 months ago

Yep, I can do this by customizing EarlyStopping.