| 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 | |