diff options
Diffstat (limited to 'dreambooth.py')
-rw-r--r-- | dreambooth.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/dreambooth.py b/dreambooth.py index a181293..db097e5 100644 --- a/dreambooth.py +++ b/dreambooth.py | |||
@@ -193,7 +193,7 @@ def parse_args(): | |||
193 | parser.add_argument( | 193 | parser.add_argument( |
194 | "--ema_power", | 194 | "--ema_power", |
195 | type=float, | 195 | type=float, |
196 | default=9 / 10 | 196 | default=5 / 6 |
197 | ) | 197 | ) |
198 | parser.add_argument( | 198 | parser.add_argument( |
199 | "--ema_max_decay", | 199 | "--ema_max_decay", |
@@ -957,7 +957,7 @@ def main(): | |||
957 | "lr/text": lr_scheduler.get_last_lr()[1] | 957 | "lr/text": lr_scheduler.get_last_lr()[1] |
958 | } | 958 | } |
959 | if args.use_ema: | 959 | if args.use_ema: |
960 | logs["ema_decay"] = ema_unet.decay | 960 | logs["ema_decay"] = 1 - ema_unet.decay |
961 | 961 | ||
962 | accelerator.log(logs, step=global_step) | 962 | accelerator.log(logs, step=global_step) |
963 | 963 | ||