seed_everything: true
ckpt_path: /app/suno/christian/boost-logs/boost/s2nfioit/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/boost-logs  # dir needs to already exist
  logger:
    class_path: pytorch_lightning.loggers.WandbLogger
    init_args:
      project: boost
      log_model: false
      save_dir: /app/suno/christian/boost-logs 
  callbacks:
    - class_path: pytorch_lightning.callbacks.ModelCheckpoint
      init_args:
        # dirpath: logs/checkpoints
        save_last: true
        save_top_k: 1
        monitor: valid_loss
        filename: "{epoch}-{step}"
    - class_path: pytorch_lightning.callbacks.ModelSummary
      init_args:
        max_depth: 2
    - class_path: pytorch_lightning.callbacks.LearningRateMonitor
    #- 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 