h2o-llmstudio icon indicating copy to clipboard operation
h2o-llmstudio copied to clipboard

[FEATURE] Support for minimum learning rate

Open tmostak opened this issue 1 year ago • 1 comments

🚀 Feature

Support for specification of a minimum learning rate

Motivation

Often in the research literature minimum learning rates are set when fine-tuning a model using a cosine or linear schedule, preventing the learning rate from dropping to 0 by the end of training. It would be very helpful if this could be supported in the LLM Studio UX.

tmostak avatar Apr 19 '24 18:04 tmostak

There is actually an open issue for that at transformers: https://github.com/huggingface/transformers/issues/28441 Once merged there, we can quickly integrate in H2O LLM Studio, as we are already using the transformers schedulers.

pascal-pfeiffer avatar May 07 '24 14:05 pascal-pfeiffer