diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

[Examples] Update train_unconditional.py to include logging argument for Wandb

Open ash0ts opened this issue 2 years ago • 1 comments

Add logger flag to choose between tensorboard and wandb for a minimal implementation Removed remainder of logging code to rediscuss for future.

Hyperlinked is a commit which will show an example utilizing an extended form of the wandb logger which can log images and model checkpoints: LINK to branch: https://github.com/ash0ts/diffusers/tree/add-wandb-unconditional-image/examples/unconditional_image_generation

Refers to this PR: https://github.com/huggingface/diffusers/pull/1245

ash0ts avatar Dec 15 '22 23:12 ash0ts

The documentation is not available anymore as the PR was closed or merged.