Unverified Commit 3dd33937 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

do not pass list to accelerator.init_trackers (#4248)

parent 5652c43f
...@@ -1116,7 +1116,9 @@ def main(args): ...@@ -1116,7 +1116,9 @@ def main(args):
# We need to initialize the trackers we use, and also store our configuration. # We need to initialize the trackers we use, and also store our configuration.
# The trackers initializes automatically on the main process. # The trackers initializes automatically on the main process.
if accelerator.is_main_process: if accelerator.is_main_process:
accelerator.init_trackers("dreambooth", config=vars(args)) tracker_config = vars(args)
tracker_config.pop("validation_images")
accelerator.init_trackers("dreambooth", config=tracker_config)
# Train! # Train!
total_batch_size = args.train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps total_batch_size = args.train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps
......
...@@ -1067,7 +1067,9 @@ def main(args): ...@@ -1067,7 +1067,9 @@ def main(args):
# We need to initialize the trackers we use, and also store our configuration. # We need to initialize the trackers we use, and also store our configuration.
# The trackers initializes automatically on the main process. # The trackers initializes automatically on the main process.
if accelerator.is_main_process: if accelerator.is_main_process:
accelerator.init_trackers("dreambooth-lora", config=vars(args)) tracker_config = vars(args)
tracker_config.pop("validation_images")
accelerator.init_trackers("dreambooth-lora", config=tracker_config)
# Train! # Train!
total_batch_size = args.train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps total_batch_size = args.train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment