search

sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 77 forks source link

[fix] pass patience arg to EarlyStopping class #152

Closed maexrakete closed 3 months ago

maexrakete commented 3 months ago

Hey mate,

you recently introduced the patience cli arg together with the EarlyStopping feature.

Unfortunately, the patience argument is never passed to the EarlyStopping class and thus will never be respected.

This PR fixes that.

Thanks for the great project and your work, cheers

sovit-123 commented 3 months ago

Thanks for the PR @maexrakete

Although it uses a default value of 10, I never pass the command line argument to the EarlyStopping class.

Merging it now.