# stage 1, 150k iters
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python main.py --base configs/latent-diffusion/brain_stage1.yaml -t --gpus 0,1,2,3,4,5,6,7,

# Edit the path-to-stage1-ckpt keyword in the stage 2 yaml
vi configs/latent-diffusion/brain_stage2.yaml

# stage 2, 50k iters
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python main.py --base configs/latent-diffusion/brain_stage2.yaml -t --gpus 0,1,2,3,4,5,6,7,




CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_stage1.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/abdomen_3d.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice_atten.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice_atten2.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice_atten3.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice_atten4.yaml -t --gpus 0,

CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/brain_slice_atten5.yaml -t --gpus 0,


CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/full_ct_2d_with_body_mask.yaml -t --gpus 0, --resume ./logs/full_ct_2d_with_body_mask

CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --base configs/latent-diffusion/full_ct_2d_with_body_mask.yaml -t --gpus 0,1,2,3, --resume ./logs/full_ct_2d_with_body_mask


torchrun --standalone --nproc_per_node=4 --master_port=29527 main.py --base configs/latent-diffusion/full_ct_2d_with_body_mask.yaml -t --gpus 0,1,2,3 --resume ./logs/full_ct_2d_with_body_mask




CUDA_VISIBLE_DEVICES=0 python main.py --base configs/latent-diffusion/full_ct_3d_with_body_mask_finetune.yaml -t --gpus 0, --resume ./logs/full_ct_3d_with_body_mask_finetune2

pip install beartype PyWavelets