Add callbacks to the model training - Githubissues

VladTomici14 / Handwritten-Recognizer

0 stars 0 forks source link

Add callbacks to the model training #2

Open VladTomici14 opened 9 months ago

VladTomici14 commented 9 months ago

Add the next features for the next model training:

timer (when it started to train, when it ended, average time / epoch)
ModelCheckpoint (saving the trained model)
graphing the evolution of the loss and valid_loss
EarlyStopping (stopping when the model doesn't evolve anymore)