File size: 1,217 Bytes
13ea048 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
{
"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
} |