kohya-ss / kohya-ss/sd-scripts
Implement Unit-Scaled Maximal Update Parametrization for best LR
- Dominant language
- Python
- Stars
- 7.2k
- Forks
- 1.2k
- Avg merge
- 11m
- Merged PRs (30d)
- 2
Description
There is a new paper, "u-μP: The Unit-Scaled Maximal Update Parametrization" at https://arxiv.org/abs/2407.17465 that promises selection of optimal hyperparameters, like the learning rate, even when using very small datatypes like FP8
The content is too advanced for me, but I had the impression that it might be a valuable addition to kohya_ss
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reading the linked u-μP paper to determine the required parametrization and whether it applies to this project. Then inspect the repository's training and optimizer entry points to identify where learning-rate and datatype behavior are configured. Done means a concrete integration plan, implementation, and validation of the claimed hyperparameter behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100