LSM / configs /model_config.yaml
kairunwen's picture
Update Code
57746f1
raw
history blame contribute delete
618 Bytes
mast3r_config:
pretrained_model_name_or_path: "checkpoints/pretrained_model/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth"
point_transformer_config:
enc_depths: [1, 1, 1, 3, 1]
enc_channels: [32, 64, 128, 256, 512]
enc_num_head: [2, 4, 8, 16, 32]
enc_patch_size: [1024, 1024, 1024, 1024, 1024]
dec_depths: [1, 1, 1, 1]
dec_channels: [64, 64, 128, 256]
dec_num_head: [4, 4, 8, 16]
dec_patch_size: [1024, 1024, 1024, 1024]
gaussian_head_config:
rgb_residual: true
d_gs_feats: 32
lseg_config:
pretrained_model_name_or_path: "checkpoints/pretrained_model/lang_seg.ckpt"
half_res: true