neggles commited on
Commit
115a6c7
1 Parent(s): cc179d9

add taming vq-gumbel-f8-8192

Browse files
taming/vq-gumbel-f8-8192/config.yaml ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.GumbelVQ
4
+ params:
5
+ kl_weight: 1.0e-08
6
+ embed_dim: 256
7
+ n_embed: 8192
8
+ monitor: val/rec_loss
9
+ temperature_scheduler_config:
10
+ target: taming.lr_scheduler.LambdaWarmUpCosineScheduler
11
+ params:
12
+ warm_up_steps: 0
13
+ max_decay_steps: 1000001
14
+ lr_start: 0.9
15
+ lr_max: 0.9
16
+ lr_min: 1.0e-06
17
+ ddconfig:
18
+ double_z: false
19
+ z_channels: 256
20
+ resolution: 256
21
+ in_channels: 3
22
+ out_ch: 3
23
+ ch: 128
24
+ ch_mult:
25
+ - 1
26
+ - 1
27
+ - 2
28
+ - 4
29
+ num_res_blocks: 2
30
+ attn_resolutions:
31
+ - 32
32
+ dropout: 0.0
33
+ lossconfig:
34
+ target: taming.modules.losses.vqperceptual.DummyLoss
taming/vq-gumbel-f8-8192/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cd6c74810ab97e00e942c25403f73afc081e8b19987b31ec0d9ff5b68e7ab14
3
+ size 376581823