File size: 933 Bytes
ae83015
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
{
    "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
    }
}