File size: 3,312 Bytes
ec14cf6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
---
license: apache-2.0
---
## Checkpoints for SlimFlow
```bash
cd SlimFlow
huggingface-cli download Yuanzhi/SlimFlow --local-dir ckpts
```
########################## CIFAR-10 ##########################
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_2_rectified_flow_100000edm_flip_warmup_300000_28m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 128 --config.model.num_res_blocks 2 --config.data.image_size 32 --config.model.ch_mult '(1, 2, 2)'
```
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_2_rectified_flow_100000edm_flip_warmup_300000_15m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 96 --config.model.num_res_blocks 2 --config.data.image_size 32 --config.model.ch_mult '(1, 2, 2)'
```
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_2_rectified_flow_100000edm_flip_warmup_300000_7m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 64 --config.model.num_res_blocks 2 --config.data.image_size 32 --config.model.ch_mult '(1, 2, 2)'
```
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_2_rectified_flow_100000edm_flip_warmup_300000_3m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 64 --config.model.num_res_blocks 1 --config.data.image_size 32 --config.model.ch_mult '(1, 1, 2)'
```
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_distill_2_rectified_flow_500000edm_lpips_use_teacher_15m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 96 --config.model.num_res_blocks 2 --config.data.image_size 32 --config.model.ch_mult '(1, 2, 2)'
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/CIFAR_distill_2_rectified_flow_500000edm_lpips_use_teacher_7m.pth" \
--config.eval.batch_size 64 \
--config.model.nf 64 --config.model.num_res_blocks 2 --config.data.image_size 32 --config.model.ch_mult '(1, 2, 2)'
```
########################## ImageNet 64 ##########################
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/ImageNet_2_rectified_flow_400000edm_flip_warmup_300000_44m.pth" \
--config.eval.batch_size 64 \
--config.model.name DhariwalUNet --config.model.nf 128 --config.model.num_res_blocks 2 --config.model.ch_mult '(1, 2, 2, 2)' --config.data.num_classes 1000 --config.data.image_size 64 --config.model.attn_resolutions '32, 16'
```
```bash
python ./image_sampling.py \
--config ./configs/rectified_flow/cifar10_rf_gaussian.py \
--sampling_dir "./ckpts/ImageNet_distill_2_rectified_flow_500000edm_lpips_use_teacher_44m.pth" \
--config.eval.batch_size 64 \
--config.model.name DhariwalUNet --config.model.nf 128 --config.model.num_res_blocks 2 --config.model.ch_mult '(1, 2, 2, 2)' --config.data.num_classes 1000 --config.data.image_size 64 --config.model.attn_resolutions '32, 16'
``` |