pip install -r requirements.txt
python test.py checkpoint_path='./diffusion.ckpt'
Training logs and checkpoints are saved inside outputs
python train.py model=TConv128Diff
python train.py model=TConv128
python train.py model=UNet
python train.py model=TConv128Spec