model:
  _target_: mayavoz.models.demucs.Demucs
  num_channels: 1
  resample: 4
  sampling_rate: 16000
  encoder_decoder:
    depth: 4
    initial_output_channels: 64
    kernel_size: 8
    stride: 4
    growth_factor: 2
    glu: true
  lstm:
    bidirectional: true
    num_layers: 2
dataset:
  _target_: mayavoz.data.dataset.MayaDataset
  root_dir: /scratch/c.sistc3/MS-SNSD/DNS20
  name: dns-2020
  duration: 5
  stride: 2
  sampling_rate: 16000
  batch_size: 32
  min_valid_minutes: 25.0
  files:
    train_clean: CleanSpeech_training
    test_clean: CleanSpeech_testing
    train_noisy: NoisySpeech_training
    test_noisy: NoisySpeech_testing
optimizer:
  _target_: torch.optim.Adam
  lr: 0.001
  betas:
  - 0.9
  - 0.999
  eps: 1.0e-08
  weight_decay: 0
  amsgrad: false
hyperparameters:
  loss: mae
  metric:
  - stoi
  - pesq
  lr: 0.0003
  ReduceLr_patience: 10
  Early_stop: false
  ReduceLr_factor: 0.5
  min_lr: 1.0e-07
  EarlyStopping_patience: 10
trainer:
  _target_: pytorch_lightning.Trainer
  accelerator: gpu
  accumulate_grad_batches: 1
  amp_backend: native
  auto_lr_find: false
  auto_scale_batch_size: false
  auto_select_gpus: true
  benchmark: false
  check_val_every_n_epoch: 1
  detect_anomaly: false
  deterministic: false
  devices: 1
  enable_checkpointing: true
  enable_model_summary: true
  enable_progress_bar: true
  fast_dev_run: false
  gpus: null
  gradient_clip_val: 0
  gradient_clip_algorithm: norm
  ipus: null
  limit_predict_batches: 1.0
  limit_test_batches: 1.0
  limit_train_batches: 1.0
  limit_val_batches: 1.0
  log_every_n_steps: 50
  max_epochs: 100
  max_steps: -1
  max_time: null
  min_epochs: 1
  min_steps: null
  move_metrics_to_cpu: false
  multiple_trainloader_mode: max_size_cycle
  num_nodes: 1
  num_processes: 1
  num_sanity_val_steps: 2
  overfit_batches: 0.0
  precision: 32
  profiler: null
  reload_dataloaders_every_n_epochs: 0
  replace_sampler_ddp: true
  strategy: null
  sync_batchnorm: false
  tpu_cores: null
  track_grad_norm: -1
  val_check_interval: 1.0
mlflow:
  experiment_name: shahules/mayavoz
  run_name: demucs-dns20