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

Make more SetFitClassification parameters configurable #46

Open
chschroeder opened this issue Aug 26, 2023 · 0 comments
Open

Make more SetFitClassification parameters configurable #46

chschroeder opened this issue Aug 26, 2023 · 0 comments
Labels
feature request New feature request

Comments

@chschroeder
Copy link
Contributor

Feature description

There are still important parameters which cannot be easily configured. The following list of parameters should be configurable via SetFitClassification / SetFitModelArguments:

  • SetFitTrainer
    • learning_rate
    • num_iterations
    • warmup_proportion

This list is intentionally not yet exhaustive, since especially with different loss functions configuration could get messy, which I want to avoid. The parameter amp would also be included, however, this is already included in #45.

Motivation

Additional comments

@chschroeder chschroeder added the feature request New feature request label Aug 26, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature request
Projects
None yet
Development

No branches or pull requests

1 participant