mally-2000 commited on
Commit
94c17d7
·
verified ·
1 Parent(s): a61817b

Delete files stage2_ldm_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. stage2_ldm_config.yaml +0 -104
stage2_ldm_config.yaml DELETED
@@ -1,104 +0,0 @@
1
- model:
2
- base_learning_rate: 2.0e-06
3
- target: ldm.models.diffusion.ddpm.LatentDiffusion
4
- params:
5
- linear_start: 0.0015
6
- linear_end: 0.0195
7
- num_timesteps_cond: 1
8
- log_every_t: 200
9
- timesteps: 1000
10
- cond_drop: 1.0
11
- first_stage_key: image
12
- use_first_stage_cond:
13
- - dipin
14
- cond_stage_key:
15
- - dipin
16
- - record
17
- cond_stage_trainable: false
18
- conditioning_key: concat
19
- concat_mode: true
20
- image_size: 64
21
- channels: 1
22
- monitor: train/loss_simple_epoch
23
- monitor_mode: min
24
- monitor_top_k: 3
25
- dynamic_size: false
26
- unet_config:
27
- target: ldm.modules.diffusionmodules.openaimodel.UNetModel
28
- params:
29
- in_channels: 6
30
- out_channels: 1
31
- model_channels: 224
32
- attention_resolutions:
33
- - 8
34
- - 4
35
- - 2
36
- num_res_blocks: 2
37
- channel_mult:
38
- - 1
39
- - 2
40
- - 3
41
- - 4
42
- num_head_channels: 32
43
- first_stage_config:
44
- target: ldm.models.autoencoder.VQModelInterface
45
- params:
46
- ckpt_path: ${oc.env:FIRST_STAGE_CKPT,./models/stage1_vqgan.ckpt}
47
- embed_dim: 1
48
- n_embed: 32768
49
- ddconfig:
50
- double_z: false
51
- z_channels: 3
52
- resolution: 256
53
- in_channels: 1
54
- out_ch: 1
55
- ch: 128
56
- ch_mult:
57
- - 1
58
- - 2
59
- - 4
60
- num_res_blocks: 2
61
- attn_resolutions: []
62
- dropout: 0
63
- lossconfig:
64
- target: torch.nn.Identity
65
- cond_stage_config:
66
- target: ldm.modules.encoders.C2f_CGuidedBlock.HWD_down4
67
- params:
68
- in_c: 1
69
- channels: 4
70
- out_c: 4
71
- wave: haar
72
- mode: reflect
73
- data:
74
- target: main.DataModuleFromConfig
75
- params:
76
- batch_size: 24
77
- num_workers: 1
78
- wrap: false
79
- train:
80
- target: ldm.data.seisimic.Marmousi3_256
81
- params:
82
- dt_path: ${oc.env:MARMousi_NPZ,./data/dtA89-1.npz}
83
- size: 256
84
- normalize: minmax
85
- ricks:
86
- - 25
87
- - 30
88
- - 35
89
- noise_snr:
90
- - 20
91
- dipins:
92
- - 0.006
93
- - 0.012
94
- - 0.024
95
- - 0.036
96
- ricks_phase:
97
- - 0
98
- train_keys:
99
- - image
100
- - dipin
101
- - record
102
- noise_type: guassian_band
103
- zhengyan_type: nonlinear
104
- record_noraml: true