wandb:
  offline: false
  project: babyhubert-finetune
  name: train
data:
  dataset_path: data/baby_train
  classes:
  - KCHI
  - OCH
  - MAL
  - FEM
  dataset_multiplier: 1.0
audio:
  chunk_duration_s: 4
  sample_rate: 16000
  strict_frames: false
model:
  name: surgical_hubert_hydra
  chkp_path: null
  config:
    wav_encoder: ""
    encoder_layers: []
    reduction: weighted
    classifier: 256
train:
  lr: 1.0e-05
  batch_size: 128
  max_epochs: 3
  validation_metric: loss
  extra_val_metrics:
  - loss
  - f1_score
  profiler: null
  dataloader:
    num_workers: 8
  scheduler:
    patience: 3