model files

config.yaml

@@ -0,0 +1,100 @@
+model:
+  _target_: enhancer.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_: enhancer.data.dataset.EnhancerDataset
+  name: vctk
+  root_dir: /scratch/c.sistc3/DS_10283_2791
+  duration: 4.5
+  stride: 0.5
+  sampling_rate: 16000
+  batch_size: 32
+  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.0003
+  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/enhancer
+  run_name: Demucs-DNS(ablation_study-noaug)

pytorch_model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18b0596f86da466ef0ecb930954ee2fee867eac19a08016655aa9ffcb894cd9a
+size 182203341