Erwann Millon commited on
Commit
f5c922b
1 Parent(s): 7b52260

change checkpoint path

Browse files
model_checkpoints/vqgan_only.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e39472bae4489764c0ffc70ba84ec7815f245781020ce55cc2e7adc60e580e4
3
+ size 288690579
model_checkpoints/vqgan_only.yaml ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ target: taming.models.vqgan.VQModel
3
+ params:
4
+ embed_dim: 256
5
+ n_embed: 1024
6
+ ddconfig:
7
+ double_z: false
8
+ z_channels: 256
9
+ resolution: 256
10
+ in_channels: 3
11
+ out_ch: 3
12
+ ch: 128
13
+ ch_mult:
14
+ - 1
15
+ - 1
16
+ - 2
17
+ - 2
18
+ - 4
19
+ num_res_blocks: 2
20
+ attn_resolutions:
21
+ - 16
22
+ dropout: 0.0
23
+ lossconfig:
24
+ target: taming.modules.losses.vqperceptual.DummyLoss
25
+ data:
26
+ target: cutlit.DataModuleFromConfig
27
+ params:
28
+ batch_size: 24
29
+ num_workers: 24
30
+ train:
31
+ target: taming.data.faceshq.CelebAHQTrain
32
+ params:
33
+ size: 256
34
+ validation:
35
+ target: taming.data.faceshq.CelebAHQValidation
36
+ params:
37
+ size: 256