{

    "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

    }

}