Emaad commited on
Commit
c27ddcd
1 Parent(s): 16950cd

Upload 2 files

Browse files
Files changed (2) hide show
  1. nucleus_vqgan.yaml +33 -0
  2. threshold_vqgan.yaml +33 -0
nucleus_vqgan.yaml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.VQModel
4
+ params:
5
+ image_key: nucleus
6
+ ckpt_path:
7
+ embed_dim: 256
8
+ n_embed: 512
9
+ ddconfig:
10
+ double_z: false
11
+ z_channels: 256
12
+ resolution: 256
13
+ in_channels: 1
14
+ out_ch: 1
15
+ ch: 128
16
+ ch_mult:
17
+ - 1
18
+ - 1
19
+ - 2
20
+ - 2
21
+ - 4
22
+ num_res_blocks: 2
23
+ attn_resolutions:
24
+ - 16
25
+ dropout: 0.0
26
+ lossconfig:
27
+ target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
28
+ params:
29
+ disc_conditional: false
30
+ disc_in_channels: 1
31
+ disc_start: 50000
32
+ disc_weight: 0.2
33
+ codebook_weight: 1.0
threshold_vqgan.yaml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-05
3
+ target: taming.models.vqgan.VQModel
4
+ params:
5
+ image_key: threshold
6
+ ckpt_path:
7
+ embed_dim: 256
8
+ n_embed: 512
9
+ ddconfig:
10
+ double_z: false
11
+ z_channels: 256
12
+ resolution: 256
13
+ in_channels: 1
14
+ out_ch: 1
15
+ ch: 128
16
+ ch_mult:
17
+ - 1
18
+ - 1
19
+ - 2
20
+ - 2
21
+ - 4
22
+ num_res_blocks: 2
23
+ attn_resolutions:
24
+ - 16
25
+ dropout: 0.0
26
+ lossconfig:
27
+ target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
28
+ params:
29
+ disc_conditional: false
30
+ disc_in_channels: 1
31
+ disc_start: 100000
32
+ disc_weight: 0.2
33
+ codebook_weight: 1.0