| { | |
| "model_type": "vae", | |
| "model_class": "pixal3d.models.autoencoders.dense_vae.DenseShapeVAE", | |
| "use_checkpoint": true, | |
| "embed_dim": 8, | |
| "in_channels": 1, | |
| "out_channels": 1, | |
| "model_channels_encoder": [ | |
| 32, | |
| 128, | |
| 512 | |
| ], | |
| "num_res_blocks_encoder": 2, | |
| "num_res_blocks_middle_encoder": 2, | |
| "model_channels_decoder": [ | |
| 512, | |
| 128, | |
| 64 | |
| ], | |
| "num_res_blocks_decoder": 4, | |
| "num_res_blocks_middle_decoder": 4, | |
| "use_fp16": true, | |
| "latents_scale": 1.0, | |
| "latents_shift": 0.0 | |
| } |