search

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.27k stars 373 forks source link

Added optional wandb logging to train.py #1

Closed johnowhitaker closed 2 years ago

johnowhitaker commented 2 years ago

Hi there, I've added logging to W&B for my own experiments, and figure it may be useful.

johnowhitaker commented 2 years ago

Example run: https://wandb.ai/johnowhitaker/k_diffusion_test/runs/23i8nsr0?workspace=user-johnowhitaker

crowsonkb commented 2 years ago

Can we get the step count on the demo grids to be the correct global step count at the time? Thank you :)

crowsonkb commented 2 years ago

I cleaned this code up/made it work with distributed training and committed/pushed my cleaned up version. Thank you very much for this! ❤️