CELL-E_2_HPA_Finetuned_480 / threshold_vqgan.yaml
Emaad's picture
Update threshold_vqgan.yaml
613249b
raw
history blame contribute delete
405 Bytes
model:
target: taming.models.vqgan.VQModel
params:
image_key: threshold
ckpt_path:
embed_dim: 256
n_embed: 512
ddconfig:
double_z: false
z_channels: 256
resolution: 256
in_channels: 1
out_ch: 1
ch: 128
ch_mult:
- 1
- 1
- 2
- 2
- 4
num_res_blocks: 2
attn_resolutions:
- 16
dropout: 0.0