Add a margin in EarlyStoppingPlugin #990
Unanswered
evertonaleixo
asked this question in
Feature Request
Replies: 1 comment 1 reply
-
Basically, something like |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I would like to add a new parameter to EarlyStoppingPlugin, called margin.
This parameter will be a float and will be used to define the minimal improvement to the plugin to accept the current step as a better model.
In some tests, the metric is improved by such a small amount, that does not really affect the accuracy model. I mean, sometimes, the improvement is just about 0.000001 in each epoch. Therefore, this parameter could reduce the training time of the experiment and still take advantage of the early stopping plugin.
I have already created a fork, implement this, and test. Do you think it's a good idea to merge into the principal repository?
Best regards,
Everton Aleixo
Beta Was this translation helpful? Give feedback.
All reactions