cfg/model/audio_diffusion.yaml

# @package _global_
model: 
  _target_: remfx.models.RemFXModel
  lr: 1e-4
  lr_beta1: 0.95
  lr_beta2: 0.999
  lr_eps: 1e-6
  lr_weight_decay: 1e-3
  sample_rate: ${sample_rate}
  network:
    _target_: remfx.models.DiffusionGenerationModel
    n_channels: 1
datamodule:
  dataset:
    effect_types: ["Clean"]
  batch_size: 2