model:
  _target_: mayavoz.models.waveunet.WaveUnet
  num_channels: 1
  depth: 9
  initial_output_channels: 24
  sampling_rate: 16000
dataset:
  _target_: mayavoz.data.dataset.MayaDataset
  name: vctk
  root_dir: /scratch/c.sistc3/DS_10283_2791
  duration: 2
  stride: 1
  sampling_rate: 16000
  batch_size: 128
  valid_minutes: 25
  files:
    train_clean: clean_trainset_28spk_wav
    test_clean: clean_testset_wav
    train_noisy: noisy_trainset_28spk_wav
    test_noisy: noisy_testset_wav
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
  - si-sdr
  lr: 0.003
  ReduceLr_patience: 10
  Early_stop: false
  ReduceLr_factor: 0.1
  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: 200
  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
  weights_save_path: null
mlflow:
  experiment_name: shahules/mayavoz
  run_name: WaveUnet-vctk