From 56958e1177455996f29c6bd2e77fe888df212652 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 6 Mar 2023 15:13:38 +0100 Subject: [PATCH] [Training] Fix tensorboard typo (#2566) --- examples/unconditional_image_generation/train_unconditional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/unconditional_image_generation/train_unconditional.py b/examples/unconditional_image_generation/train_unconditional.py index b46b4edb36..e9c849b22b 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -648,7 +648,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