Skip to content
This repository was archived by the owner on Apr 8, 2025. It is now read-only.

EarlyStopping fix and custom head#617

Merged
Timoeller merged 2 commits intodeepset-ai:masterfrom
ftesser:early_stopping_fix_and_custom_head
Nov 6, 2020
Merged

EarlyStopping fix and custom head#617
Timoeller merged 2 commits intodeepset-ai:masterfrom
ftesser:early_stopping_fix_and_custom_head

Conversation

@ftesser
Copy link
Copy Markdown
Contributor

@ftesser ftesser commented Nov 5, 2020

Hello, this PR, contains two commit related to EarlyStopping class:

  • the first fix a bug with the return in a particular case
  • the second is an improvement that allow the user to choose the prediction head to refer to for the metric (useful for Natural Questions).

@Timoeller Timoeller self-requested a review November 6, 2020 10:54
@Timoeller Timoeller self-assigned this Nov 6, 2020
Copy link
Copy Markdown
Contributor

@Timoeller Timoeller left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looking good.

Thanks for creating the new PR with only relevant changes.
Nice bug fix catch there btw and you also added a docstring for the head. I wish all external PRs would be like that : )

@Timoeller Timoeller merged commit 9b6a29d into deepset-ai:master Nov 6, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants