| { | |
| "embed_dim": 4, | |
| "monitor": "val/rec_loss", | |
| "ddconfig": { | |
| "double_z": true, | |
| "z_channels": 4, | |
| "resolution": 256, | |
| "in_channels": 3, | |
| "out_ch": 3, | |
| "ch": 128, | |
| "ch_mult": [ | |
| 1, | |
| 2, | |
| 4, | |
| 4 | |
| ], | |
| "num_res_blocks": 2, | |
| "attn_resolutions": [], | |
| "dropout": 0.0 | |
| }, | |
| "lossconfig": { | |
| "target": "torch.nn.Identity" | |
| }, | |
| "_target": "crs_core.autoencoder.AutoencoderKL" | |
| } |