finetune_scan_objbg.yaml

optimizer : {
  type: AdamW,
  kwargs: {
  lr : 0.0005,
  weight_decay : 0.05
  }}

scheduler: {
  type: CosLR,
  kwargs: {
    epochs: 300,
    initial_epochs : 10
  }}

dataset : {
  train : { _base_: cfgs/dataset_configs/ScanObjectNN_objectbg.yaml,
            others: {subset: 'train'}},
  val : { _base_: cfgs/dataset_configs/ScanObjectNN_objectbg.yaml,
            others: {subset: 'test'}},
  test : { _base_: cfgs/dataset_configs/ScanObjectNN_objectbg.yaml,
            others: {subset: 'test'}}}
model : {
  NAME: PointMamba,
  trans_dim: 384,
  depth: 12,
  cls_dim: 15,
  num_heads: 6,
  group_size: 32,
  num_group: 128,
  encoder_dims: 384,
  rms_norm: False,
  drop_path: 0.1,
  drop_out: 0.,
}


npoints: 2048
total_bs : 32
step_per_update : 1
max_epoch : 300
grad_norm_clip : 10