update ae config
This commit is contained in:
parent
dd10f44cdb
commit
c02c06b778
1 changed files with 1 additions and 0 deletions
|
@ -49,6 +49,7 @@ model:
|
||||||
monitor: val/rec_loss
|
monitor: val/rec_loss
|
||||||
ckpt_path: "models/first_stage_models/kl-f16/model.ckpt"
|
ckpt_path: "models/first_stage_models/kl-f16/model.ckpt"
|
||||||
ddconfig:
|
ddconfig:
|
||||||
|
double_z: True
|
||||||
z_channels: 16
|
z_channels: 16
|
||||||
resolution: 256
|
resolution: 256
|
||||||
in_channels: 3
|
in_channels: 3
|
||||||
|
|
Loading…
Reference in a new issue