-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
72 changed files
with
4,743 additions
and
50 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,14 @@ | ||
| torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p train data.batch_size=8 | ||
| torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p train | ||
| # | ||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/metfaces_base.yaml -p train data.batch_size=8 | ||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/metfaces_base.yaml -p train | ||
|
|
||
|
|
||
|
|
||
| torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/faster/train/checkpoints/0040000.pt" | ||
|
|
||
| torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/base/train/checkpoints/0040000.pt" | ||
|
|
||
|
|
||
| # eval | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/faster/inference/0040000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/base/inference/0040000 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1 @@ | ||
| torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/image_256/dit_xl2.yaml -p train | ||
| 0torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/image_256/dit_xl2.yaml -p train |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,23 @@ | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/faster/train/checkpoints/0040000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/base/train/checkpoints/0040000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/faster/train/checkpoints/0030000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/base/train/checkpoints/0030000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/faster/train/checkpoints/0020000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/base/train/checkpoints/0020000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/faster/train/checkpoints/0010000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces/base/train/checkpoints/0010000.pt" | ||
|
|
||
|
|
||
|
|
||
| # eval | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/faster/inference/0040000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/base/inference/0040000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/faster/inference/0030000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/base/inference/0030000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/faster/inference/0020000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/base/inference/0020000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/faster/inference/0010000 | ||
| python -m pytorch_fid /home/yuanzhihang/metfaces/images /mnt/public/yuanzhihang/outputs/metfaces/base/inference/0010000 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,27 @@ | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p train | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p train | ||
| # | ||
| #python main.py -c configs/image/unconditional/ffhq_base.yaml -p train ckpt_path="" | ||
|
|
||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0070000.pt" | ||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0070000.pt" | ||
|
|
||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0080000.pt" | ||
| torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0080000.pt" | ||
|
|
||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0060000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0060000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0030000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0030000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0020000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0020000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0010000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0010000.pt" | ||
| # | ||
|
|
||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/faster/train/checkpoints/0010000.pt" | ||
| #torchrun --nproc_per_node=8 --master_port=30002 main.py -c configs/image/unconditional/ffhq_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/ffhq/base/train/checkpoints/0010000.pt" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,17 @@ | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0080000 | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0080000 | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0070000 | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0070000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0040000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0040000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0030000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0030000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0020000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0020000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/faster/inference/0010000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/ffhq/images256x256 /mnt/public/yuanzhihang/outputs/ffhq/base/inference/0010000 | ||
|
|
||
|
|
||
| # 6:21 | ||
| torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p train | ||
| torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_base.yaml -p train |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,6 @@ | ||
| torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/unconditional/metfaces_theory.yaml -p train | ||
|
|
||
|
|
||
| python main.py -c configs/image/text2img/mscoco_faster.yaml -p sample ckpt_path=/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0140000.pt | ||
|
|
||
| python main.py -c configs/image/text2img/mscoco_base.yaml -p sample ckpt_path=/mnt/public/yuanzhihang/outputs/mscoco/real_base/train/checkpoints/0140000.pt |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,25 @@ | ||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0100000.pt" | ||
|
|
||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0100000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0200000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0300000.pt" | ||
| # | ||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0400000.pt" | ||
|
|
||
| #torchrun --nproc_per_node=8 --master_port=30003 main.py -c configs/image/text2img/mscoco_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/real_base/train/checkpoints/0100000.pt" | ||
|
|
||
| #python main.py -c configs/image/text2img/mscoco_base.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/real_base/train/checkpoints/0100000.pt" | ||
| # | ||
| #python main.py -c configs/image/text2img/mscoco_faster.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0100000.pt" | ||
| #python main.py -c configs/image/text2img/mscoco_faster.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0200000.pt" | ||
| #python main.py -c configs/image/text2img/mscoco_faster.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0300000.pt" | ||
| #python main.py -c configs/image/text2img/mscoco_faster.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/mscoco/base/train/checkpoints/0400000.pt" | ||
|
|
||
| #python -m pytorch_fid /mnt/public/yuanzhihang/mscoco/val2017 /mnt/public/yuanzhihang/outputs/mscoco/base/inference/0100000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/mscoco/val2017 /mnt/public/yuanzhihang/outputs/mscoco/base/inference/0200000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/mscoco/val2017 /mnt/public/yuanzhihang/outputs/mscoco/base/inference/0300000 | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/mscoco/val2017 /mnt/public/yuanzhihang/outputs/mscoco/base/inference/0400000 | ||
|
|
||
| #python -m pytorch_fid /mnt/public/yuanzhihang/mscoco/val /mnt/public/yuanzhihang/outputs/mscoco/real_base/inference/0100000 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,30 @@ | ||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_512/faster.yaml -p train | ||
|
|
||
| #torchrun --nproc_per_node=8 --master_port=30001 main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p train | ||
|
|
||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0020000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0040000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0060000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0080000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0100000.pt" | ||
|
|
||
|
|
||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0020000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0040000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0060000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0080000.pt" | ||
| #python main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p sample ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0100000.pt" | ||
|
|
||
| #torchrun --nproc_per_node=8 main.py -c configs/image/unconditional/metfaces_512/faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0040000.pt" | ||
| #torchrun --nproc_per_node=8 main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0040000.pt" | ||
| # | ||
| # | ||
| #torchrun --nproc_per_node=8 main.py -c configs/image/unconditional/metfaces_512/faster.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/faster/train/checkpoints/0100000.pt" | ||
| #torchrun --nproc_per_node=8 main.py -c configs/image/unconditional/metfaces_512/baseline.yaml -p inference ckpt_path="/mnt/public/yuanzhihang/outputs/metfaces_512/base/train/checkpoints/0100000.pt" | ||
|
|
||
| #python -m pytorch_fid /mnt/public/yuanzhihang/outputs/metfaces_512/faster/inference/0040000 /mnt/public/yuanzhihang/metfaces | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/outputs/metfaces_512/faster/inference/0100000 /mnt/public/yuanzhihang/metfaces | ||
| #python -m pytorch_fid /mnt/public/yuanzhihang/outputs/metfaces_512/base/inference/0040000 /mnt/public/yuanzhihang/metfaces | ||
| python -m pytorch_fid /mnt/public/yuanzhihang/outputs/metfaces_512/base/inference/0100000 /mnt/public/yuanzhihang/metfaces | ||
|
|
||
| #python |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,2 @@ | ||
| torchrun --nproc_per_node=8 main.py -c configs/image/imagenet_256/mdt/baseline.yaml -p train | ||
| torchrun --nproc_per_node=8 main.py -c configs/image/imagenet_256/mdt/faster.yaml -p train |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| #FFHQ_DIR="" | ||
| #METFACES_DIR="" | ||
|
|
||
| torchrun --nproc_per_node=8 main.py -c configs/comparison/metfaces/p2.yaml -p train | ||
| torchrun --nproc_per_node=8 main.py -c configs/comparison/metfaces/min_snr.yaml -p train | ||
|
|
||
| torchrun --nproc_per_node=8 main.py -c configs/comparison/ffhq/p2.yaml -p train | ||
| torchrun --nproc_per_node=8 main.py -c configs/comparison/ffhq/min_snr.yaml -p train | ||
|
|
||
|
|
||
| # data in: /home/zyk/metfaces/images |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,43 @@ | ||
| image_size: 256 | ||
| experiment_name: unconditional | ||
|
|
||
| optimizer: | ||
| _target_: torch.optim.AdamW | ||
| lr: 0.0001 | ||
| weight_decay: 0 | ||
|
|
||
|
|
||
| vae: | ||
| _target_: diffusers.models.AutoencoderKL.from_pretrained | ||
| pretrained_model_name_or_path: "transformers/sd-vae-ft-ema" | ||
|
|
||
|
|
||
| data: | ||
| dataset: | ||
| _target_: speedit.dataset.image.image_dataset | ||
| image_size: ${image_size} | ||
| class_cond: false | ||
|
|
||
| batch_size: 32 | ||
| num_workers: 4 | ||
|
|
||
| model: | ||
| _target_: speedit.networks.dit.DiT_XL_2 | ||
| condition: none | ||
|
|
||
|
|
||
| sample: | ||
| diffusion: | ||
| timestep_respacing: '250' | ||
|
|
||
| inference: | ||
| diffusion: | ||
| timestep_respacing: '250' | ||
| per_proc_batch_size: 32 | ||
| num_samples: 10000 | ||
|
|
||
| epoch: 200_000 | ||
| max_training_steps: 50_000 | ||
|
|
||
| log_every: 100 | ||
| ckpt_every: 10_000 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| data: | ||
| dataset: | ||
| root: /mnt/public/yuanzhihang/ffhq/images256x256/ | ||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.iddpm.IDDPM | ||
| timestep_respacing: "" | ||
|
|
||
| experiment_dir: /mnt/public/yuanzhihang/outputs/ffhq/base |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: outputs/ffhq/clts | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: none | ||
| sampling: CLTS |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: outputs/ffhq/lognorm | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: none | ||
| sampling: lognorm |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: outputs/ffhq/min_snr | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: min_snr | ||
| sampling: uniform |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: outputs/ffhq/p2 | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: p2 | ||
| sampling: uniform |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| data: | ||
| dataset: | ||
| root: /mnt/public/yuanzhihang/metfaces | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.iddpm.IDDPM | ||
| timestep_respacing: "" | ||
|
|
||
| experiment_dir: /mnt/public/yuanzhihang/outputs/metfaces/base |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: /mnt/public/yuanzhihang/outputs/metfaces/clts | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: none | ||
| sampling: CLTS |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: /mnt/public/yuanzhihang/outputs/metfaces/lognorm | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: none | ||
| sampling: lognorm |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: /mnt/public/yuanzhihang/outputs/metfaces/min_snr | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: min_snr | ||
| sampling: uniform |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| experiment_dir: /mnt/public/yuanzhihang/outputs/metfaces/p2 | ||
|
|
||
|
|
||
| diffusion: | ||
| _target_: speedit.diffusion.speed.Speed_IDDPM | ||
| timestep_respacing: "" | ||
| weighting: p2 | ||
| sampling: uniform |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.