summaryrefslogtreecommitdiffstats
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rw-r--r--training/functional.py2
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(