cifar10_vae / config.json
kohido's picture
Push model using huggingface_hub.
27efd35 verified
{
"conv_type": "conv2d",
"dec_act_func": "GLU",
"dec_act_kwargs": {
"dim": 1
},
"dim": 128,
"enc_act_func": "LeakyReLU",
"enc_act_kwargs": {
"negative_slope": 0.1
},
"first_conv_kernel_size": 5,
"group": 4,
"in_channel": 3,
"l2_recon_loss": true,
"layer_mults": [
1,
1,
1,
1
],
"layers": 4,
"num_res_blocks": 1,
"out_channel": 3,
"use_variational": true
}