DDPM (Denoising Diffusion Probabilistic Models)

pip3 install -U pip setuptools
pip3 install -r requirements.txt
pip3 install protobuf==3.20.3
yum install -y mesa-libGL 
pip3 install urllib3==1.26.6

mkdir -p stats && cd stats

stats
└── cifar10.train.npz

cd ..

# 8 GPUs
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

python3 main.py --train \
    --flagfile ./config/CIFAR10.txt \
    --parallel

# 1 GPU
export CUDA_VISIBLE_DEVICES=0

python3 main.py --train \
    --flagfile ./config/CIFAR10.txt

# 8 GPUs
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

python3 main.py \
    --flagfile ./logs/DDPM_CIFAR10_EPS/flagfile.txt \
    --notrain \
    --eval \
    --parallel

# 1 GPU
export CUDA_VISIBLE_DEVICES=0

python3 main.py \
    --flagfile ./logs/DDPM_CIFAR10_EPS/flagfile.txt \
    --notrain \
    --eval