| { | |
| "target": "hy3dshape.models.autoencoders.shape_vae_pl.ShapeVAEPL", | |
| "params": { | |
| "module_cfg": { | |
| "target": "hy3dshape.models.autoencoders.shape_vae.ShapeVAE", | |
| "params": { | |
| "num_latents": 4096, | |
| "train_num_latents_range": [ | |
| 4096 | |
| ], | |
| "train_num_latents_probs": [ | |
| 1.0 | |
| ], | |
| "downsample_ratio": 20, | |
| "embed_dim": 64, | |
| "point_feats": 3, | |
| "sharpedge_label": true, | |
| "pc_size": 81920, | |
| "pc_sharpedge_size": 0, | |
| "num_freqs": 8, | |
| "include_pi": false, | |
| "heads": 16, | |
| "width": 1024, | |
| "num_encoder_layers": 8, | |
| "num_decoder_layers": 16, | |
| "use_ln_post": true, | |
| "init_scale": 0.25, | |
| "qkv_bias": false, | |
| "flash": true, | |
| "qk_norm": true, | |
| "use_checkpoint": false | |
| } | |
| }, | |
| "loss_cfg": { | |
| "target": "torch.nn.Identity" | |
| } | |
| }, | |
| "scale_factor": 1.0079887561230991 | |
| } |