Skip to content

Repository files navigation

Diffusion Guidance project

Installation

  • conda create -n ddpm python=3.10.10
  • conda activate ddpm
  • conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
  • pip install -r requirements.txt

Logging

  • The code uses tensorboard to log the train loss. Use the command tensorboard --logdir=runs to observe the training loss.
  • To train a classifier-free guidance model run python ddmp_train.py --cfg

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages