From cdeb73ccda731206226dd83d76a5783f5d441831 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 6 Mar 2023 13:34:43 +0000 Subject: [PATCH] [Training] Fix tensorboard typo --- 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 b46b4edb363b..e9c849b22b6b 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -648,7 +648,7 @@ def transform_images(examples): 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