diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/functional.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py index a9c7a8a..e1035ce 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -556,7 +556,7 @@ def train( | |||
| 556 | global_step_offset: int = 0, | 556 | global_step_offset: int = 0, |
| 557 | with_prior_preservation: bool = False, | 557 | with_prior_preservation: bool = False, |
| 558 | prior_loss_weight: float = 1.0, | 558 | prior_loss_weight: float = 1.0, |
| 559 | low_freq_noise: float = 0.05, | 559 | low_freq_noise: float = 0.1, |
| 560 | **kwargs, | 560 | **kwargs, |
| 561 | ): | 561 | ): |
| 562 | text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, extra = strategy.prepare( | 562 | text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, extra = strategy.prepare( |