Update README.md

This commit is contained in:
Xavier 2022-09-06 09:35:58 -07:00 committed by GitHub
parent 05d2a195a8
commit 0c9611f251
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -33,4 +33,7 @@ python main.py --base configs/stable-diffusion/v1-finetune_unfrozen.yaml
--class_word <xxx> --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 ### Generation