diff options
| author | XavierXiao <xiaozhisheng950@gmail.com> | 2022-09-06 09:32:32 -0700 |
|---|---|---|
| committer | XavierXiao <xiaozhisheng950@gmail.com> | 2022-09-06 09:32:32 -0700 |
| commit | 05d2a195a827e2721b6a3c3fb48e00c2fd69fe4d (patch) | |
| tree | dda206ccb19cf836a395b7e966b8ae23e1358ac2 | |
| parent | Update README.md (diff) | |
change lr
| -rw-r--r-- | configs/stable-diffusion/v1-finetune_unfrozen.yaml | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/configs/stable-diffusion/v1-finetune_unfrozen.yaml b/configs/stable-diffusion/v1-finetune_unfrozen.yaml index 0c1409a..2f85316 100644 --- a/configs/stable-diffusion/v1-finetune_unfrozen.yaml +++ b/configs/stable-diffusion/v1-finetune_unfrozen.yaml @@ -1,5 +1,5 @@ model: - base_learning_rate: 1.0e-05 + base_learning_rate: 1.0e-06 target: ldm.models.diffusion.ddpm.LatentDiffusion params: reg_weight: 1.0 @@ -19,7 +19,7 @@ model: use_ema: False embedding_reg_weight: 0.0 unfreeze_model: True - model_lr: 1.0e-5 + model_lr: 1.0e-6 personalization_config: target: ldm.modules.embedding_manager.EmbeddingManager |