Training: fix tensorboard tracking

Crashed with accelerate < 0.17.0.dev0
2025-12-06 20:44:33 +08:00 · 2023-03-03 10:58:37 +01:00
1 changed files with 1 additions and 1 deletions
--- a/examples/unconditional_image_generation/train_unconditional.py
+++ b/examples/unconditional_image_generation/train_unconditional.py
@@ -631,7 +631,7 @@ def main(args):
                    if is_accelerate_version(">=", "0.17.0.dev0"):
                        tracker = accelerator.get_tracker("tensorboard", unwrap=True)
                    else:
-                        tracker = accelerator.get_tracker()
+                        tracker = accelerator.get_tracker("tensorboard")
                    tracker.add_images("test_samples", images_processed.transpose(0, 3, 1, 2), epoch)
                elif args.logger == "wandb":
                    # Upcoming `log_images` helper coming in https://github.com/huggingface/accelerate/pull/962/files