From 0c9611f25142808a0ec507714abd1f5a8af2f257 Mon Sep 17 00:00:00 2001 From: Xavier Date: Tue, 6 Sep 2022 09:35:58 -0700 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) 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 ``` +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