cba_mlp.yaml

seml:
  name: trn_mlp
  executable: exp_train.py
  project_root_dir: ../../
  output_dir: log/training

slurm:
  experiments_per_job: 1
  sbatch_options:
    gres: gpu:1       # num GPUs
    mem: 10G          # memory
    cpus-per-task: 4  # num cores
    time: 0-00:30     # max time, D-HH:MM

fixed:
  data_params:
    graph_model: CBA
    m: 2
    
  train_params:
    inductive: True

  model_params:
    label: MLP
    model: MLP

  other_params:
    sacred_metrics: False

grid:
  data_params:
    K:
      type: choice
      options:
        - 0.1
        - 0.5
        - 1
        - 1.5
        - 2
        - 3
        - 4
        - 5

  model_params:
    hidden_channels:
      type: choice
      options:
        - 32
        - 64
        - 128
        - 256

    dropout:
      type: choice
      options:
        - 0.0
        - 0.3
        - 0.5

  train_params:
    lr: 
      type: choice
      options:
        - 0.1
        - 0.01
        - 0.001

    weight_decay:
      type: choice
      options:
        - 0.01
        - 0.001
        - 0.0001

  seed:
    type: range
    min: 0
    max: 10
    step: 1