Early stopping - Githubissues

alecokas / swahili-text-gcn

Graph Convolutional Network for Swahili News Classification: https://arxiv.org/abs/2103.09325

https://arxiv.org/abs/2103.09325

MIT License

8 stars 4 forks source link

Closed tyler-martin-12 closed 3 years ago

tyler-martin-12 commented 3 years ago

Added CLI args for disable_early_stopping (default is False) and early_stopping_epochs, which is 10 by default
In the training loop, I added check for the number of epochs since the metric of interest has improved
I tested this on a dummy example and it worked as expected
Added a function for saving training notes, which is used if early stopping is used to break the training loop