# core
core:
  version: "musicfm_concat_k_mood"
  task: "mood"
  precision: "32-true" # "32-true" or "16-mixed" or "bf16-mixed"
  save_top_k: 1
  devices: 1
  num_nodes: 1
  max_epochs: 200

data:
  batch_size: 32
  val_batch_size: 32
  num_workers: 4
  dataset: "deam"
  limit_train: 1.0

optim:
  learning_rate: 1e-4

model:
  frontend: "musicfm_concat_k4096"
  backend: "2-layer"
  latent_dim: 1024
  output_dim: 2
  layer_ix: 7
  is_flash: False
  output_size: -1
