Update

1 files changed, 3 insertions, 3 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 280cf77..6d699f3 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -87,7 +87,7 @@ def parse_args():
     parser.add_argument(
         "--num_buckets",
         type=int,
-        default=4,
+        default=0,
         help="Number of aspect ratio buckets in either direction.",
     )
     parser.add_argument(
@@ -305,7 +305,7 @@ def parse_args():
     parser.add_argument(
         "--adam_weight_decay",
         type=float,
-        default=0,
+        default=1e-2,
         help="Weight decay to use."
     )
     parser.add_argument(
@@ -526,6 +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,
     )
 
     checkpoint_output_dir = output_dir / "model"
@@ -587,7 +588,6 @@ def main():
         seed=args.seed,
         optimizer=optimizer,
         lr_scheduler=lr_scheduler,
-        prepare_unet=True,
         num_train_epochs=args.num_train_epochs,
         gradient_accumulation_steps=args.gradient_accumulation_steps,
         sample_frequency=args.sample_frequency,