Skip to content

Fix evaluation KeyError for "eval_one_score" and model loading issue for ModernBERT#227

Closed
ianbulovic wants to merge 3 commits intoMachine-Learning-for-Medical-Language:mainfrom
ianbulovic:fix-eval-one-score
Closed

Fix evaluation KeyError for "eval_one_score" and model loading issue for ModernBERT#227
ianbulovic wants to merge 3 commits intoMachine-Learning-for-Medical-Language:mainfrom
ianbulovic:fix-eval-one-score

Conversation

@ianbulovic
Copy link
Copy Markdown
Contributor

@ianbulovic ianbulovic commented Apr 22, 2025

This fixes the following error during evaluation: KeyError: "The `metric_for_best_model` training argument is set to 'eval_one_score', which is not found in the evaluation metrics.

Looking back at the discussion in #190, there was some concern about saving the model twice. Not sure if this is still an issue...

This also fixes an issue that was causing the final best model load at the end of training to fail for ModernBERT fine-tuning.

@ianbulovic ianbulovic requested a review from tmills April 22, 2025 14:34
@ianbulovic ianbulovic changed the title Fix evaluation KeyError for "eval_one_score" Fix evaluation KeyError for "eval_one_score" and model loading issue for ModernBERT Apr 22, 2025
@ianbulovic
Copy link
Copy Markdown
Contributor Author

Tests are passing but ModernBERT is broken again...

@tmills tmills closed this Apr 23, 2025
@tmills
Copy link
Copy Markdown
Contributor

tmills commented Apr 23, 2025

superseded by #228

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants