Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The trainer class of the medkit library lack of more specific options #19

Open
Thibeb opened this issue Feb 9, 2024 · 0 comments
Open
Labels
enhancement New feature or request

Comments

@Thibeb
Copy link
Contributor

Thibeb commented Feb 9, 2024

When fine-tuning a BERT model using the medkit library I had to compare it to a fine-tuning process made directly with the hugging face library and I stood upon the fact that I couldn't replicate exactly what was done with the HF library in medkit.

Options such as weight decay and or early stopping could not be implemented.

I don't know if it would be possible to have more access to the hugging face training library directly trough the medkit-lib but it would help having more options when training models.

@ghisvail ghisvail added the enhancement New feature or request label Feb 19, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants