| { | |
| "ae": { | |
| "model_id": "wfvae", | |
| "from_pretrained": null, | |
| "base_channels": 128, | |
| "decoder_energy_flow_hidden_size": 128, | |
| "decoder_num_resblocks": 2, | |
| "dropout": 0.0, | |
| "encoder_energy_flow_hidden_size": 128, | |
| "encoder_num_resblocks": 2, | |
| "latent_dim": 4, | |
| "use_attention": true, | |
| "norm_type": "aelayernorm", | |
| "t_interpolation": "trilinear", | |
| "use_tiling": false, | |
| "connect_res_layer_num": 1, | |
| "training": true | |
| }, | |
| "discriminator": { | |
| "model_id": "LPIPSWithDiscriminator3D", | |
| "perceptual_from_pretrained": ".cache/lpips/vgg.pth", | |
| "discrim_start": 0, | |
| "kl_weight": 1e-6, | |
| "discrim_weight": 0.5, | |
| "logvar_init": 0.0, | |
| "perceptual_weight": 1.0, | |
| "loss_type": "l1", | |
| "discrim_loss": "hinge", | |
| "use_dropout": true | |
| } | |
| } |