| { | |
| "aeconfig": { | |
| "z_channels": 4, | |
| "resolution": 128, | |
| "in_channels": 96, | |
| "out_ch": 96, | |
| "ch": 256, | |
| "ch_mult": [4, 4, 2, 1], | |
| "res_updown_level": [4], | |
| "num_res_blocks": 2, | |
| "dropout": 0.1 | |
| }, | |
| "embed_dim": 4, | |
| "latent_scale_factor": null, | |
| "kl_weight": 1e-06, | |
| "geom_weight": 1.0, | |
| "recon_loss_type": "l2" | |
| } |