From e32b4d4c04a31b22051740e5f26e16960464f787 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 3 Mar 2023 18:53:15 +0100 Subject: Implemented different noise offset --- train_dreambooth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index 6d699f3..8571dff 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -438,7 +438,7 @@ def main(): accelerator = Accelerator( log_with=LoggerType.TENSORBOARD, - logging_dir=f"{output_dir}", + project_dir=f"{output_dir}", mixed_precision=args.mixed_precision ) @@ -526,7 +526,7 @@ def main(): with_prior_preservation=args.num_class_images != 0, prior_loss_weight=args.prior_loss_weight, no_val=args.valid_set_size == 0, - # low_freq_noise=0, + # noise_offset=0, ) checkpoint_output_dir = output_dir / "model" -- cgit v1.2.3-54-g00ecf