Add early stopping and pruning to hparam search - Githubissues

sorenhartmann / mlops_project

MIT License

2 stars 0 forks source link

Add early stopping and pruning to hparam search #24

Closed sorenhartmann closed 3 years ago

MikkelMathiasen23 commented 3 years ago

Early stopping PL module: https://pytorch-lightning.readthedocs.io/en/latest/common/early_stopping.html Pruning: https://optuna.readthedocs.io/en/v1.0.0/tutorial/pruning.html

Integrate optuna earlystopping callback in pytorch lightning: https://optuna.readthedocs.io/en/v1.0.0/reference/integration.html#optuna.integration.PyTorchLightningPruningCallback

idariis commented 3 years ago

Early stopping has been implemented but not pruning. We will look at this Tuesday if the model is still overfitting.