scannet_nano.yaml

# @package _global_

# to execute this experiment run:
# python train.py experiment=semantic/scannet_nano

defaults:
  - override /datamodule: semantic/scannet_nano.yaml
  - override /model: semantic/nano-2.yaml
  - override /trainer: gpu.yaml

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

trainer:
  max_epochs: 100
  check_val_every_n_epoch: 2

model:
  optimizer:
    lr: 0.01
    weight_decay: 1e-4

  scheduler:
    num_warmup: 2

  _node_mlp_out: 32
  _h_edge_mlp_out: 32
  _down_dim: [ 32, 32, 32, 32 ]
  _up_dim: [ 32, 32, 32 ]
  net:
    no_ffn: False
    down_ffn_ratio: 1

logger:
  wandb:
    project: "spt_scannet"
    name: "NANO"