seed: 3407

dataset:
  name: 'CoordinatesData'
  args:
    data_root: '/Users/zhouweiqi/Downloads/gcfp/data/dataset'
    train_anno_file: '/Users/zhouweiqi/Downloads/gcfp/data/dataset/train.csv'
    val_anno_file: '/Users/zhouweiqi/Downloads/gcfp/data/dataset/valid.csv'

dataloader:
  batch_size: 32
  num_workers: 4
  pin_memory: true
  shuffle: true

model:
  name: 'MLPModel'
  args:
    activation: 'relu'

solver:
  name: 'MLPSolver'
  args:
    epoch: 100

  optimizer:
    name: 'Adam'
    args:
      lr: !!float 1e-4
      weight_decay: !!float 5e-5

  lr_scheduler:
    name: 'StepLR'
    args:
      step_size: 15
      gamma: 0.1

  loss:
    name: 'SigmoidFocalLoss'
    # name: 'CrossEntropyLoss'
    args:
      reduction: "mean"

  logger:
    log_root: '/Users/zhouweiqi/Downloads/test/logs'
    suffix: 'mlp'