seed_everything: true
ckpt_path: /app/suno/christian/ear-logs/ear/0bwk1wd7/checkpoints/last.ckpt 

trainer:
  precision: 32
  devices: 1
  num_sanity_val_steps: 2
  accumulate_grad_batches: 1
  strategy: ddp_find_unused_parameters_true
  accelerator: gpu
  default_root_dir: /app/suno/christian/ear-logs  # dir needs to already exist
  log_every_n_steps: 1
  logger:
    class_path: pytorch_lightning.loggers.WandbLogger
    init_args:
      project: ear
      log_model: false
      save_dir: /app/suno/christian/ear-logs 
  callbacks:
    - class_path: pytorch_lightning.callbacks.ModelCheckpoint
      init_args:
        # dirpath: logs/checkpoints
        save_last: true
        save_top_k: 1
        monitor: valid_mean_acc
        mode: max
        filename: "{epoch}-{step}"
    - class_path: pytorch_lightning.callbacks.ModelSummary
      init_args:
        max_depth: 2
    - class_path: pytorch_lightning.callbacks.LearningRateMonitor
    #- class_path: suno_ear.callbacks.EvaluateCallback
    #  init_args:
    #    real_audios_dir: /app/suno/christian/reference-audio-wav
    #    generated_audios_dir: /app/suno/christian/data/v3_generations_10k
    #    num_compare: 10
    #s    num_frames: 131072
    #- class_path: boost.callbacks.InferAudioFromDirectory
    #  init_args:
    #    root_dir: /app/suno/christian/suno_generations
    #    num_examples: 8
    #    shuffle: false
    #    norm_type: loudness
    #    num_steps: 50
    #- class_path: boost.callbacks.InferAudioFromDirectory
    #  init_args:
    #    root_dir: /app/suno/christian/data/codec_audio/reference-audio-wav/dac_2c_25x12
    #    num_examples: 16
    #    shuffle: false
    #    norm_type: loudness
    #    num_steps: 50
    #- class_path: lcap.callbacks.MoveConfigCallback
    #- class_path: lcap.callbacks.ConfusionMatrixCallback
  # enable_checkpointing: true 