{
  "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"
}