Update config file
Browse files
ddpm_ip_cifar10/config-2023-08-19-07-26-49.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_ip.DDPM_IP
|
| 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_ip.DDPM_IP
|
| 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
|