config.json

{
  "model": {
    "type": "PoLArMAE-SemSeg",
    "num_classes": 4,
    "arch": "vit_small",
    "voxel_size": 5.0,
    "num_channels": 4,
    "seg_head_fetch_layers": [
      3,
      7,
      11
    ],
    "seg_head_combination_method": "mean",
    "seg_head_dim": 512,
    "seg_head_dropout": 0.5,
    "freeze_encoder": false,
    "apply_encoder_postnorm": false,
    "condition_global_features": true,
    "upsampling_k": 5,
    "center": [
      384.0,
      384.0,
      384.0
    ],
    "scale": 0.0015035163260146505,
    "transformer_kwargs": {
      "postnorm": false,
      "add_pos_at_every_layer": true,
      "drop_rate": 0.0,
      "attn_drop_rate": 0.05,
      "drop_path_rate": 0.25
    },
    "criteria": [
      {
        "type": "CrossEntropyLoss",
        "loss_weight": 1.0,
        "ignore_index": -1
      },
      {
        "type": "LovaszLoss",
        "mode": "multiclass",
        "loss_weight": 0.05,
        "ignore_index": -1
      }
    ]
  }
}