LitePT#

class LitePT(in_channels=4, order=('z', 'z-trans', 'hilbert', 'hilbert-trans'), stride=(2, 2, 2, 2), enc_depths=(2, 2, 2, 6, 2), enc_channels=(36, 72, 144, 252, 504), enc_num_head=(2, 4, 8, 14, 28), enc_patch_size=(1024, 1024, 1024, 1024, 1024), enc_conv=(True, True, True, False, False), enc_attn=(False, False, False, True, True), enc_rope_freq=(100.0, 100.0, 100.0, 100.0, 100.0), dec_depths=(0, 0, 0, 0), dec_channels=(72, 72, 144, 252), dec_num_head=(4, 4, 8, 14), dec_patch_size=(1024, 1024, 1024, 1024), dec_conv=(False, False, False, False), dec_attn=(False, False, False, False), dec_rope_freq=(100.0, 100.0, 100.0, 100.0), mlp_ratio=4, qkv_bias=True, qk_scale=None, attn_drop=0.0, proj_drop=0.0, drop_path=0.3, pre_norm=True, shuffle_orders=True, mask_token=False, enc_mode=False, freeze_encoder=False, traceable=True)[source]#: Bases: PointModule