# core
core:
  version: "mert_lyric"
  task: "ctc"
  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: 24
  val_batch_size: 24
  num_workers: 4
  dataset: "genius_ctc"
  limit_train: 1.0

optim:
  learning_rate: 1e-4

model:
  frontend: "mert_330m"
  backend: "blstm"
  latent_dim: 1024
  output_dim: 74
  layer_ix: 7
  is_flash: False
  output_size: 750