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'
```