-
Notifications
You must be signed in to change notification settings - Fork 0
/
cdiff_config.json
85 lines (85 loc) · 2.1 KB
/
cdiff_config.json
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
79
80
81
82
83
84
85
{
"name": "tcw",
"phase": "train",
"gpu_ids": [1],
"path": {
"log": "logs",
"tb_logger": "tb_logger",
"results": "results",
"checkpoint": "checkpoint",
"resume_state": null
},
"datasets": {
"train": {
"name": "TCW",
"mode": "HR",
"dataroot": "data/era5_tcw/train",
"datatype": "lmdb",
"l_resolution": 16,
"r_resolution": 128,
"batch_size": 32,
"num_workers": 8,
"use_shuffle": true,
"data_len": -1
},
"val": {
"name": "TCW",
"mode": "LRHR",
"dataroot": "data/era5_tcw/train",
"datatype": "lmdb",
"l_resolution": 128,
"r_resolution": 128,
"data_len": 50
}
},
"model": {
"which_model_G": "cddpm",
"finetune_norm": false,
"unet": {
"in_channel": 2,
"out_channel": 1,
"inner_channel": 16,
"channel_multiplier": [1, 2, 4, 8, 8],
"attn_res": [8],
"res_blocks": 1,
"dropout": 0
},
"beta_schedule": {
"train": {
"schedule": "cosine",
"n_timestep": 2000,
"linear_start": 1e-6,
"linear_end": 1e-2
},
"val": {
"schedule": "cosine",
"n_timestep": 10,
"linear_start": 1e-6,
"linear_end": 1e-2
}
},
"diffusion": {
"image_size": 128,
"channels": 1,
"conditional": true
}
},
"train": {
"n_iter": 150000,
"val_freq": 2e4,
"save_checkpoint_freq": 2e4,
"print_freq": 2000,
"optimizer": {
"type": "adam",
"lr": 1e-4
},
"ema_scheduler": {
"step_start_ema": 5000,
"update_ema_every": 1,
"ema_decay": 0.9999
}
},
"wandb": {
"project": "sr_ffhq"
}
}