File size: 426 Bytes
83b7193 |
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 |
vae:
use_downsample: true
num_latents: 256
point_feats: 3
out_dim: 1
embed_dim: 64
width: 768
heads: 12
num_encoder_layers: 8
num_decoder_layers: 16
init_scale: 0.25
qkv_bias: false
use_ln_post: true
use_udf_extraction: true
token_scales:
- 128.0
- 256.0
- 384.0
- 512.0
- 640.0
- 1024.0
- 2048.0
token_probability:
- 0.025
- 0.025
- 0.025
- 0.025
- 0.05
- 0.2
- 0.65
|