python train_diffusion.py --sampling ddpm --model waves --wb -n 100 --lr_schedule --lr 0.001 --prediction_type v_prediction
python train_diffusion.py --sampling ddpm --model life --wb -n 100 --lr_schedule --lr 0.001 --prediction_type v_prediction
python train_diffusion.py --sampling ddpm --model maze --wb -n 100 --lr_schedule --lr 0.001 --prediction_type v_prediction
python train_diffusion.py --sampling ddpm --model spirals --wb -n 100 --lr_schedule --lr 0.001 --prediction_type v_prediction