Skip to content

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

Description

@Thibeb

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.

Activity

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions