diff options
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, |