diff options
author | Volpeon <git@volpeon.ink> | 2023-01-16 15:52:43 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-01-16 15:52:43 +0100 |
commit | 6c8cffe28baeafac77d047ff3f8ded9418033e2f (patch) | |
tree | 807c527deb1b15ef795f5cd8a7682151c69a037e /training/strategy | |
parent | Pad dataset if len(items) < batch_size (diff) | |
download | textual-inversion-diff-6c8cffe28baeafac77d047ff3f8ded9418033e2f.tar.gz textual-inversion-diff-6c8cffe28baeafac77d047ff3f8ded9418033e2f.tar.bz2 textual-inversion-diff-6c8cffe28baeafac77d047ff3f8ded9418033e2f.zip |
More training adjustments
Diffstat (limited to 'training/strategy')
-rw-r--r-- | training/strategy/ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/strategy/ti.py b/training/strategy/ti.py index 568f9eb..9d39e15 100644 --- a/training/strategy/ti.py +++ b/training/strategy/ti.py | |||
@@ -36,7 +36,7 @@ def textual_inversion_strategy( | |||
36 | use_emb_decay: bool = False, | 36 | use_emb_decay: bool = False, |
37 | emb_decay_target: float = 0.4, | 37 | emb_decay_target: float = 0.4, |
38 | emb_decay_factor: float = 1, | 38 | emb_decay_factor: float = 1, |
39 | emb_decay_start: float = 1e-4, | 39 | emb_decay_start: float = 0, |
40 | use_ema: bool = False, | 40 | use_ema: bool = False, |
41 | ema_inv_gamma: float = 1.0, | 41 | ema_inv_gamma: float = 1.0, |
42 | ema_power: int = 1, | 42 | ema_power: int = 1, |