diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-27 10:30:43 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-27 10:30:43 +0200 |
| commit | bd903c95053d2b5c4a85475e9ce3a2037b40d92a (patch) | |
| tree | de0d52394ea62a69073432dbad12cbf999a64de3 /dreambooth.py | |
| parent | New Euler_a scheduler (diff) | |
| download | textual-inversion-diff-bd903c95053d2b5c4a85475e9ce3a2037b40d92a.tar.gz textual-inversion-diff-bd903c95053d2b5c4a85475e9ce3a2037b40d92a.tar.bz2 textual-inversion-diff-bd903c95053d2b5c4a85475e9ce3a2037b40d92a.zip | |
Euler_a: Re-introduce generator arg for reproducible output
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 | ||