Optimization of training model by val_loss/epoch#24
Conversation
|
@andruuhurst I have a question regarding this PR to improve my understanding of what is happening. It seems like you are finding the minimum loss value and saving the epoch number where the I believe they should be set to |
|
@David-Rod That is a good question. We have found our ideal amount of epochs from setting aside the validation set initially. When retraining the model it is no longer necessary to set aside a validation set since we were only using that metric to gage underfitting/overfitting of the model. From my understanding, this allows us to instead use the entirety of the training data on the model and in a way, it can theoretically a see more full range of data to be trained on |
|
The master branch will need to be merged with this branch to avoid merge conflicts. Several changes have been merged since this PR was requested. |
David-Rod
left a comment
There was a problem hiding this comment.
Branch not up to date with current changes. Need to merge with master and remove commented print statements.
Implemented the retraining of the model with the best epoch by finding the lowest validation loss value from model history.
In my experience at work and in the CS499 deep learning class, this is the best practice for optimizing the training model.
I also doubled the number of epochs trained for the initial model. Realistically, this will not affect the final trained model. I did this to prevent the potential possibility of GD having the case of underfitting with only 20 epochs since they could possibly get different results with their data.
Let me know what you guys think.