kohya-ss / kohya-ss/sd-scripts

Implement Unit-Scaled Maximal Update Parametrization for best LR

Open
#1,439 0 comments 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.