model: | |
target: pixel_generator.vec2face.taming.models.vqgan.VQModel | |
params: | |
embed_dim: 256 | |
n_embed: 1024 | |
ddconfig: | |
double_z: False | |
z_channels: 256 | |
resolution: 112 | |
in_channels: 3 | |
out_ch: 3 | |
ch: 128 | |
ch_mult: [ 1,1,2,2,4] # num_down = len(ch_mult)-1 | |
num_res_blocks: 2 | |
attn_resolutions: [16] | |
dropout: 0.0 | |