aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorXavier <xiaozhisheng950@gmail.com>2022-09-06 09:35:58 -0700
committerGitHub <noreply@github.com>2022-09-06 09:35:58 -0700
commit0c9611f25142808a0ec507714abd1f5a8af2f257 (patch)
treee744c51ef7f77b348a175e689d8c26cafdcdce2a
parentchange lr (diff)
Update README.md
-rw-r--r--README.md3
1 files changed, 3 insertions, 0 deletions
diff --git a/README.md b/README.md
index 7133ca3..13c1b3c 100644
--- a/README.md
+++ b/README.md
@@ -33,4 +33,7 @@ python main.py --base configs/stable-diffusion/v1-finetune_unfrozen.yaml
--class_word <xxx>
```
+Detailed configuration can be found in configs/stable-diffusion/v1-finetune_unfrozen.yaml. In particular, the default learning rate is ```1.0e-6``` as I found the ```1.0e-5``` in the Dreambooth paper leads to poor editability. The parameter ```reg_weight``` corresponds to the weight of regularization in the Dreambooth paper, and the default is set to ```1.0```.
+
+
### Generation