Skip to content

text_encoder_lr_scale_step#40

Open
maruo555 wants to merge 1 commit intomaruo-mainfrom
feature/text_encoder_lr_scale_step
Open

text_encoder_lr_scale_step#40
maruo555 wants to merge 1 commit intomaruo-mainfrom
feature/text_encoder_lr_scale_step

Conversation

@maruo555
Copy link
Owner

--text_encoder_lr_scale_step,--text_encoder_lr_scale_factor

指定したグローバル step 到達時にText Encoderの学習率を倍率で一度だけ変更。
TE1/TE2の個別LR指定がある場合は、それぞれの元LRに倍率を適用。

train_network.py の TextEncoderLrScaleManager で制御。
skip_grad_norm による更新スキップが発生しても学習率変更は一度だけ適用される。
入力制約: scale_step は 0 以上の整数、scale_factor は 0 より大きい実数。片方だけ指定するとエラーで停止。

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant