| { | |
| "codebook_size": 4096, | |
| "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": null, | |
| "layers": 1, | |
| "num_res_blocks": 0, | |
| "out_channel": 3, | |
| "quantizer": "LFQ", | |
| "quantizer_kwargs": { | |
| "diversity_gamma": 1.0, | |
| "entropy_loss_weight": 0.02, | |
| "experimental_softplus_entropy_loss": true, | |
| "num_codebooks": 8, | |
| "spherical": true | |
| } | |
| } |