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

data:
  batch_size: 64
  val_batch_size: 1
  num_workers: 4
  dataset: "giantstep"
  limit_train: 1.0

optim:
  learning_rate: 1e-4

model:
  frontend: "musicfm_msd16"
  backend: "2-layer"
  latent_dim: 1024
  output_dim: 24
  layer_ix: 7
  is_flash: True
  output_size: -1