| { | |
| "library_name": "torch-pointcloud", | |
| "name": "litept-base.structured3d.prs-eth", | |
| "task": "semantic-segmentation", | |
| "hparams": { | |
| "in_channels": 6, | |
| "num_classes": 25, | |
| "serialization_orders": [ | |
| "z", | |
| "z-trans", | |
| "hilbert", | |
| "hilbert-trans" | |
| ], | |
| "shuffle_serialization_orders": true, | |
| "strides": [ | |
| 2, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "encoder_depths": [ | |
| 3, | |
| 3, | |
| 3, | |
| 12, | |
| 3 | |
| ], | |
| "encoder_channels": [ | |
| 54, | |
| 108, | |
| 216, | |
| 432, | |
| 576 | |
| ], | |
| "encoder_num_heads": [ | |
| 3, | |
| 6, | |
| 12, | |
| 24, | |
| 32 | |
| ], | |
| "encoder_patch_size": [ | |
| 1024, | |
| 1024, | |
| 1024, | |
| 1024, | |
| 1024 | |
| ], | |
| "encoder_conv": [ | |
| true, | |
| true, | |
| true, | |
| false, | |
| false | |
| ], | |
| "encoder_attention": [ | |
| false, | |
| false, | |
| false, | |
| true, | |
| true | |
| ], | |
| "decoder_depths": [ | |
| 0, | |
| 0, | |
| 0, | |
| 0 | |
| ], | |
| "decoder_channels": [ | |
| 432, | |
| 216, | |
| 108, | |
| 72 | |
| ], | |
| "decoder_num_heads": [ | |
| 24, | |
| 12, | |
| 6, | |
| 4 | |
| ], | |
| "decoder_patch_size": [ | |
| 1024, | |
| 1024, | |
| 1024, | |
| 1024 | |
| ], | |
| "decoder_conv": [ | |
| false, | |
| false, | |
| false, | |
| false | |
| ], | |
| "decoder_attention": [ | |
| false, | |
| false, | |
| false, | |
| false | |
| ], | |
| "norm": "batch_norm", | |
| "norm_kwargs": { | |
| "eps": 0.001, | |
| "momentum": 0.01 | |
| }, | |
| "act": "gelu", | |
| "mlp_ratio": 4.0, | |
| "drop_path": 0.3, | |
| "rope_base": 100.0, | |
| "use_flash_attn": true | |
| }, | |
| "input_keys": [ | |
| "x", | |
| "pos_grid", | |
| "batch" | |
| ], | |
| "dataset": "structured3d", | |
| "classes": [ | |
| "wall", | |
| "floor", | |
| "cabinet", | |
| "bed", | |
| "chair", | |
| "sofa", | |
| "table", | |
| "door", | |
| "window", | |
| "picture", | |
| "desk", | |
| "shelves", | |
| "curtain", | |
| "dresser", | |
| "pillow", | |
| "mirror", | |
| "ceiling", | |
| "refrigerator", | |
| "television", | |
| "nightstand", | |
| "sink", | |
| "lamp", | |
| "otherstructure", | |
| "otherfurniture", | |
| "otherprop" | |
| ], | |
| "author": "prs-eth", | |
| "license": "MIT" | |
| } | |