| { |
| "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 |
| } |
