# core
core:
  version: "ditto_artist_ytmsd_128_continue"
  precision: "bf16-mixed" # "32-true" or "16-mixed" or "bf16-mixed"
  save_top_k: 2
  devices: 8
  num_nodes: 1
  max_epochs: 999999999
  ckpt_path: null

# data
data:
  train_dataset: "artist_ytmsd"
  valid_dataset: "artist_ytmsd"
  batch_size: 10
  num_workers: 16
  limit_train: 1.0
  num_train_samples: -1
  num_val_samples: -1

# optimization
optim:
  learning_rate: 1e-4

# model parameters
model:
  version: "artist_ytmsd"
  music_encoder_name: "musicfm_concat"
  text_encoder_name: "xlm-roberta"
  model_path: "/home/minz/logs/ditto_cleaned_ytmsd_128/step_370k.pt"
  latent_dim: 128