Guanxiong commited on
Commit
810a83e
·
1 Parent(s): 3529555

add vqvae and cardiac model

Browse files
Autoencoder/vqvae/vq_cardiac_d4.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2975a8c8aa564ab49c0cbcaacddac7aa19d68af208f3f00183e75d3dbff6c5d
3
+ size 756213966
Autoencoder/vqvae/vq_cardiac_d4.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.VQModel
4
+ params:
5
+ embed_dim: 3
6
+ n_embed: 8192
7
+ ddconfig:
8
+ double_z: false
9
+ z_channels: 3
10
+ resolution: 256
11
+ in_channels: 3
12
+ out_ch: 3
13
+ ch: 128
14
+ ch_mult:
15
+ - 1
16
+ - 2
17
+ - 4
18
+ num_res_blocks: 2
19
+ attn_resolutions: []
20
+ dropout: 0.0
21
+ lossconfig:
22
+ target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
23
+ params:
24
+ disc_conditional: false
25
+ disc_in_channels: 3
26
+ disc_start: 30001
27
+ disc_weight: 0.8
28
+ codebook_weight: 1.0
29
+ data:
30
+ target: main.DataModuleFromConfig
31
+ params:
32
+ batch_size: 6
33
+ num_workers: 24
34
+ train:
35
+ target: taming.data.imagenet.ImageNetSR
36
+ params:
37
+ config:
38
+ size: 256
39
+ validation:
40
+ target: taming.data.imagenet.ImageNetSRVal
41
+ params:
42
+ config:
43
+ size: 256