Commit
•
e49a0d6
1
Parent(s):
e47f707
remove config
Browse files- config.yaml +0 -100
config.yaml
DELETED
|
@@ -1,100 +0,0 @@
|
|
| 1 |
-
model:
|
| 2 |
-
_target_: enhancer.models.demucs.Demucs
|
| 3 |
-
num_channels: 1
|
| 4 |
-
resample: 4
|
| 5 |
-
sampling_rate: 16000
|
| 6 |
-
encoder_decoder:
|
| 7 |
-
depth: 4
|
| 8 |
-
initial_output_channels: 64
|
| 9 |
-
kernel_size: 8
|
| 10 |
-
stride: 4
|
| 11 |
-
growth_factor: 2
|
| 12 |
-
glu: true
|
| 13 |
-
lstm:
|
| 14 |
-
bidirectional: true
|
| 15 |
-
num_layers: 2
|
| 16 |
-
dataset:
|
| 17 |
-
_target_: enhancer.data.dataset.EnhancerDataset
|
| 18 |
-
name: vctk
|
| 19 |
-
root_dir: /scratch/c.sistc3/DS_10283_2791
|
| 20 |
-
duration: 4.5
|
| 21 |
-
stride: 0.5
|
| 22 |
-
sampling_rate: 16000
|
| 23 |
-
batch_size: 32
|
| 24 |
-
valid_minutes: 25
|
| 25 |
-
files:
|
| 26 |
-
train_clean: clean_trainset_28spk_wav
|
| 27 |
-
test_clean: clean_testset_wav
|
| 28 |
-
train_noisy: noisy_trainset_28spk_wav
|
| 29 |
-
test_noisy: noisy_testset_wav
|
| 30 |
-
optimizer:
|
| 31 |
-
_target_: torch.optim.Adam
|
| 32 |
-
lr: 0.001
|
| 33 |
-
betas:
|
| 34 |
-
- 0.9
|
| 35 |
-
- 0.999
|
| 36 |
-
eps: 1.0e-08
|
| 37 |
-
weight_decay: 0
|
| 38 |
-
amsgrad: false
|
| 39 |
-
hyperparameters:
|
| 40 |
-
loss: mae
|
| 41 |
-
metric:
|
| 42 |
-
- stoi
|
| 43 |
-
- pesq
|
| 44 |
-
- si-sdr
|
| 45 |
-
lr: 0.0003
|
| 46 |
-
ReduceLr_patience: 10
|
| 47 |
-
Early_stop: false
|
| 48 |
-
ReduceLr_factor: 0.1
|
| 49 |
-
min_lr: 1.0e-07
|
| 50 |
-
EarlyStopping_patience: 10
|
| 51 |
-
trainer:
|
| 52 |
-
_target_: pytorch_lightning.Trainer
|
| 53 |
-
accelerator: gpu
|
| 54 |
-
accumulate_grad_batches: 1
|
| 55 |
-
amp_backend: native
|
| 56 |
-
auto_lr_find: false
|
| 57 |
-
auto_scale_batch_size: false
|
| 58 |
-
auto_select_gpus: true
|
| 59 |
-
benchmark: false
|
| 60 |
-
check_val_every_n_epoch: 1
|
| 61 |
-
detect_anomaly: false
|
| 62 |
-
deterministic: false
|
| 63 |
-
devices: 1
|
| 64 |
-
enable_checkpointing: true
|
| 65 |
-
enable_model_summary: true
|
| 66 |
-
enable_progress_bar: true
|
| 67 |
-
fast_dev_run: false
|
| 68 |
-
gpus: null
|
| 69 |
-
gradient_clip_val: 0
|
| 70 |
-
gradient_clip_algorithm: norm
|
| 71 |
-
ipus: null
|
| 72 |
-
limit_predict_batches: 1.0
|
| 73 |
-
limit_test_batches: 1.0
|
| 74 |
-
limit_train_batches: 1.0
|
| 75 |
-
limit_val_batches: 1.0
|
| 76 |
-
log_every_n_steps: 50
|
| 77 |
-
max_epochs: 200
|
| 78 |
-
max_steps: -1
|
| 79 |
-
max_time: null
|
| 80 |
-
min_epochs: 1
|
| 81 |
-
min_steps: null
|
| 82 |
-
move_metrics_to_cpu: false
|
| 83 |
-
multiple_trainloader_mode: max_size_cycle
|
| 84 |
-
num_nodes: 1
|
| 85 |
-
num_processes: 1
|
| 86 |
-
num_sanity_val_steps: 2
|
| 87 |
-
overfit_batches: 0.0
|
| 88 |
-
precision: 32
|
| 89 |
-
profiler: null
|
| 90 |
-
reload_dataloaders_every_n_epochs: 0
|
| 91 |
-
replace_sampler_ddp: true
|
| 92 |
-
strategy: null
|
| 93 |
-
sync_batchnorm: false
|
| 94 |
-
tpu_cores: null
|
| 95 |
-
track_grad_norm: -1
|
| 96 |
-
val_check_interval: 1.0
|
| 97 |
-
weights_save_path: null
|
| 98 |
-
mlflow:
|
| 99 |
-
experiment_name: shahules/enhancer
|
| 100 |
-
run_name: Demucs-DNS(ablation_study-noaug)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|