Update config file
Browse files
ddpm_pred_x0_cifar10/config-2023-08-07-07-49-01.yaml
CHANGED
@@ -1,15 +1,15 @@
|
|
1 |
seed: 2022
|
2 |
data:
|
3 |
-
|
4 |
-
|
5 |
-
|
|
|
6 |
img_channels: 3
|
7 |
num_classes: 10
|
8 |
dataloader:
|
9 |
num_workers: 4
|
10 |
pin_memory: true
|
11 |
prefetch_factor: 2
|
12 |
-
micro_batch: 0
|
13 |
model:
|
14 |
target: models.unet.UNet
|
15 |
params:
|
@@ -29,8 +29,6 @@ model:
|
|
29 |
num_res_blocks: 2
|
30 |
n_heads: 1
|
31 |
dropout: 0.1
|
32 |
-
ema_decay: 0.9999
|
33 |
-
ema_gradual: true
|
34 |
diffusion:
|
35 |
target: diffusions.ddpm.DDPM
|
36 |
params:
|
@@ -43,7 +41,10 @@ diffusion:
|
|
43 |
train:
|
44 |
n_steps: 800000
|
45 |
batch_size: 128
|
|
|
46 |
clip_grad_norm: 1.0
|
|
|
|
|
47 |
print_freq: 400
|
48 |
save_freq: 10000
|
49 |
sample_freq: 5000
|
|
|
1 |
seed: 2022
|
2 |
data:
|
3 |
+
target: datasets.cifar10.CIFAR10
|
4 |
+
params:
|
5 |
+
root: /data/fengxin/xuyifeng/data/CIFAR-10/
|
6 |
+
img_size: 32
|
7 |
img_channels: 3
|
8 |
num_classes: 10
|
9 |
dataloader:
|
10 |
num_workers: 4
|
11 |
pin_memory: true
|
12 |
prefetch_factor: 2
|
|
|
13 |
model:
|
14 |
target: models.unet.UNet
|
15 |
params:
|
|
|
29 |
num_res_blocks: 2
|
30 |
n_heads: 1
|
31 |
dropout: 0.1
|
|
|
|
|
32 |
diffusion:
|
33 |
target: diffusions.ddpm.DDPM
|
34 |
params:
|
|
|
41 |
train:
|
42 |
n_steps: 800000
|
43 |
batch_size: 128
|
44 |
+
micro_batch: 0
|
45 |
clip_grad_norm: 1.0
|
46 |
+
ema_decay: 0.9999
|
47 |
+
ema_gradual: true
|
48 |
print_freq: 400
|
49 |
save_freq: 10000
|
50 |
sample_freq: 5000
|
ddpm_pred_x0_cifar10/config-2023-08-09-23-34-33.yaml
CHANGED
@@ -1,15 +1,15 @@
|
|
1 |
seed: 2022
|
2 |
data:
|
3 |
-
|
4 |
-
|
5 |
-
|
|
|
6 |
img_channels: 3
|
7 |
num_classes: 10
|
8 |
dataloader:
|
9 |
num_workers: 4
|
10 |
pin_memory: true
|
11 |
prefetch_factor: 2
|
12 |
-
micro_batch: 0
|
13 |
model:
|
14 |
target: models.unet.UNet
|
15 |
params:
|
@@ -29,8 +29,6 @@ model:
|
|
29 |
num_res_blocks: 2
|
30 |
n_heads: 1
|
31 |
dropout: 0.1
|
32 |
-
ema_decay: 0.9999
|
33 |
-
ema_gradual: true
|
34 |
diffusion:
|
35 |
target: diffusions.ddpm.DDPM
|
36 |
params:
|
@@ -43,7 +41,10 @@ diffusion:
|
|
43 |
train:
|
44 |
n_steps: 800000
|
45 |
batch_size: 128
|
|
|
46 |
clip_grad_norm: 1.0
|
|
|
|
|
47 |
print_freq: 400
|
48 |
save_freq: 10000
|
49 |
sample_freq: 5000
|
|
|
1 |
seed: 2022
|
2 |
data:
|
3 |
+
target: datasets.cifar10.CIFAR10
|
4 |
+
params:
|
5 |
+
root: /data/fengxin/xuyifeng/data/CIFAR-10/
|
6 |
+
img_size: 32
|
7 |
img_channels: 3
|
8 |
num_classes: 10
|
9 |
dataloader:
|
10 |
num_workers: 4
|
11 |
pin_memory: true
|
12 |
prefetch_factor: 2
|
|
|
13 |
model:
|
14 |
target: models.unet.UNet
|
15 |
params:
|
|
|
29 |
num_res_blocks: 2
|
30 |
n_heads: 1
|
31 |
dropout: 0.1
|
|
|
|
|
32 |
diffusion:
|
33 |
target: diffusions.ddpm.DDPM
|
34 |
params:
|
|
|
41 |
train:
|
42 |
n_steps: 800000
|
43 |
batch_size: 128
|
44 |
+
micro_batch: 0
|
45 |
clip_grad_norm: 1.0
|
46 |
+
ema_decay: 0.9999
|
47 |
+
ema_gradual: true
|
48 |
print_freq: 400
|
49 |
save_freq: 10000
|
50 |
sample_freq: 5000
|