conda create -n ddpm python=3.10.10conda activate ddpmconda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidiapip install -r requirements.txt
- The code uses tensorboard to log the train loss. Use the command
tensorboard --logdir=runsto observe the training loss. - To train a classifier-free guidance model run python ddmp_train.py --cfg