Skip to content

GitLab

Unverified Commit 56958e11 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Training] Fix tensorboard typo (#2566)

parent ec021923
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/unconditional_image_generation/train_unconditional.py
View file @ 56958e11
...@@ -648,7 +648,7 @@ def main(args): ...@@ -648,7 +648,7 @@ def main(args):
if is_accelerate_version(">=", "0.17.0.dev0"): if is_accelerate_version(">=", "0.17.0.dev0"):
tracker = accelerator.get_tracker("tensorboard", unwrap=True) tracker = accelerator.get_tracker("tensorboard", unwrap=True)
else: else:
tracker = accelerator.get_tracker() tracker = accelerator.get_tracker("tensorboard")
tracker.add_images("test_samples", images_processed.transpose(0, 3, 1, 2), epoch) tracker.add_images("test_samples", images_processed.transpose(0, 3, 1, 2), epoch)
elif args.logger == "wandb": elif args.logger == "wandb":
# Upcoming `log_images` helper coming in https://github.com/huggingface/accelerate/pull/962/files # Upcoming `log_images` helper coming in https://github.com/huggingface/accelerate/pull/962/files
......
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