Aaai25 paper, experimental code. We will further improve the code in the future.
This is the codebase for Diffusion Models Beat GANS on Image Synthesis.
From the link, download the checkpoint "ffhq_10m.pt" and paste it to ./weights/
cd scripts python -m torch.distributed.launch --nproc_per_node=1 --use_env --master_port=29522 sr_sample.py --model_path [pre_trained_path] --CRT_path [CRT_path] --h_sets [GT/Real-World Path] --downsampling 8 --out_path [out_path]
cd scripts python -m torch.distributed.launch --nproc_per_node=1 --use_env --master_port=29522 image_train.py --data_dir [Your_training_sets] --batch_size 32 --lr_anneal_steps 100000