hparams.yaml

config:
  seed: 64736289
  name: default
  train: true
  test: false
  path:
    exp_root: exp
    datasets:
      wsj0_mix: data/wsj0_mix
    figures: figures
  datamodule:
    train:
      dl_opts:
        num_workers: 8
        shuffle: true
        batch_size: 6
      dataset:
        _target_: datasets.WSJ0_mix
        path: data/wsj0_mix
        n_spkr: 2
        fs: 8000
        cut: max
        split: train
        max_len_s: 5
        max_n_samples: null
    val:
      dl_opts:
        num_workers: 8
        shuffle: false
        batch_size: 5
      dataset:
        _target_: datasets.WSJ0_mix
        path: data/wsj0_mix
        n_spkr: 2
        fs: 8000
        cut: max
        split: val
        max_len_s: null
        max_n_samples: null
    test:
      dl_opts:
        num_workers: 8
        shuffle: false
        batch_size: 5
      dataset:
        _target_: datasets.WSJ0_mix
        path: data/wsj0_mix
        n_spkr: 2
        fs: 8000
        cut: max
        split: test
        max_len_s: null
        max_n_samples: null
  model:
    n_speakers: 2
    fs: 8000
    t_eps: 0.03
    t_rev_init: 0.03
    ema_decay: 0.999
    valid_max_sep_batches: 2
    time_sampling_strategy: uniform
    train_source_order: power
    init_hack: 5
    mmnr_thresh_pit: -10.0
    score_model:
      _target_: models.score_models.ScoreModelNCSNpp
      num_sources: 2
      stft_args:
        n_fft: 510
        hop_length: 128
        center: true
        pad_mode: constant
      backbone_args:
        _target_: models.ncsnpp.NCSNpp
        nf: 128
      transform: exponent
      spec_abs_exponent: 0.5
      spec_factor: 0.15
    sde:
      _target_: sdes.sdes.MixSDE
      ndim: 2
      d_lambda: 2.0
      sigma_min: 0.05
      sigma_max: 0.5
      'N': 30
    sampler:
      'N': 30
      snr: 0.5
      corrector_steps: 1
    loss:
      _target_: torch.nn.MSELoss
    main_val_loss: val/si_sdr
    main_val_loss_mode: max
    val_losses:
      val/si_sdr:
        _target_: models.losses.SISDRLoss
        zero_mean: true
        clamp_db: 30
        reduction: mean
        sign_flip: true
    optimizer:
      _target_: torch.optim.Adam
      lr: 0.0002
      weight_decay: 0.0
    scheduler: null
    grad_clipper:
      _target_: utils.FixedClipper
      max_norm: 5.0
    init_hack_p: 0.1
  trainer:
    _target_: pytorch_lightning.Trainer
    accumulate_grad_batches: 2
    min_epochs: 1
    max_epochs: 1000
    deterministic: true
    accelerator: gpu
    devices: -1
    strategy: ddp
    auto_select_gpus: true
    check_val_every_n_epoch: 5
    default_root_dir: .
    profiler: false