niobures commited on
Commit
0803d0e
·
verified ·
1 Parent(s): 2d548ca

Demucs (models)

Browse files
models/mayavoz-demucs-ms-snsd-20/.gitattributes ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tflite filter=lfs diff=lfs merge=lfs -text
29
+ *.tgz filter=lfs diff=lfs merge=lfs -text
30
+ *.wasm filter=lfs diff=lfs merge=lfs -text
31
+ *.xz filter=lfs diff=lfs merge=lfs -text
32
+ *.zip filter=lfs diff=lfs merge=lfs -text
33
+ *.zst filter=lfs diff=lfs merge=lfs -text
34
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
models/mayavoz-demucs-ms-snsd-20/README.md ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
models/mayavoz-demucs-ms-snsd-20/config.yaml ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ _target_: mayavoz.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_: mayavoz.data.dataset.MayaDataset
18
+ root_dir: /scratch/c.sistc3/MS-SNSD/DNS20
19
+ name: dns-2020
20
+ duration: 5
21
+ stride: 2
22
+ sampling_rate: 16000
23
+ batch_size: 32
24
+ min_valid_minutes: 25.0
25
+ files:
26
+ train_clean: CleanSpeech_training
27
+ test_clean: CleanSpeech_testing
28
+ train_noisy: NoisySpeech_training
29
+ test_noisy: NoisySpeech_testing
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
+ lr: 0.0003
45
+ ReduceLr_patience: 10
46
+ Early_stop: false
47
+ ReduceLr_factor: 0.5
48
+ min_lr: 1.0e-07
49
+ EarlyStopping_patience: 10
50
+ trainer:
51
+ _target_: pytorch_lightning.Trainer
52
+ accelerator: gpu
53
+ accumulate_grad_batches: 1
54
+ amp_backend: native
55
+ auto_lr_find: false
56
+ auto_scale_batch_size: false
57
+ auto_select_gpus: true
58
+ benchmark: false
59
+ check_val_every_n_epoch: 1
60
+ detect_anomaly: false
61
+ deterministic: false
62
+ devices: 1
63
+ enable_checkpointing: true
64
+ enable_model_summary: true
65
+ enable_progress_bar: true
66
+ fast_dev_run: false
67
+ gpus: null
68
+ gradient_clip_val: 0
69
+ gradient_clip_algorithm: norm
70
+ ipus: null
71
+ limit_predict_batches: 1.0
72
+ limit_test_batches: 1.0
73
+ limit_train_batches: 1.0
74
+ limit_val_batches: 1.0
75
+ log_every_n_steps: 50
76
+ max_epochs: 100
77
+ max_steps: -1
78
+ max_time: null
79
+ min_epochs: 1
80
+ min_steps: null
81
+ move_metrics_to_cpu: false
82
+ multiple_trainloader_mode: max_size_cycle
83
+ num_nodes: 1
84
+ num_processes: 1
85
+ num_sanity_val_steps: 2
86
+ overfit_batches: 0.0
87
+ precision: 32
88
+ profiler: null
89
+ reload_dataloaders_every_n_epochs: 0
90
+ replace_sampler_ddp: true
91
+ strategy: null
92
+ sync_batchnorm: false
93
+ tpu_cores: null
94
+ track_grad_norm: -1
95
+ val_check_interval: 1.0
96
+ mlflow:
97
+ experiment_name: shahules/mayavoz
98
+ run_name: demucs-dns20
models/mayavoz-demucs-ms-snsd-20/pytorch_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52f6c4fcb0dbe32e03ae55ac09a3715d875dd928be5a7fd72accd9523d919409
3
+ size 182201101
models/mayavoz-demucs-ms-snsd-20/source.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://huggingface.co/shahules786/mayavoz-demucs-ms-snsd-20
models/mayavoz-demucs-valentini-28spk/.gitattributes ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tflite filter=lfs diff=lfs merge=lfs -text
29
+ *.tgz filter=lfs diff=lfs merge=lfs -text
30
+ *.wasm filter=lfs diff=lfs merge=lfs -text
31
+ *.xz filter=lfs diff=lfs merge=lfs -text
32
+ *.zip filter=lfs diff=lfs merge=lfs -text
33
+ *.zst filter=lfs diff=lfs merge=lfs -text
34
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
models/mayavoz-demucs-valentini-28spk/README.md ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
models/mayavoz-demucs-valentini-28spk/config.yml ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ _target_: mayavoz.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_: mayavoz.data.dataset.MayaDataset
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/mayavoz
100
+ run_name: Demucs-DNS(ablation_study-noaug)
models/mayavoz-demucs-valentini-28spk/pytorch_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0830d0642426dd93a8c81f7903c28a62c26a5cdbb574cb69f91866b66c85fc7c
3
+ size 182201293
models/mayavoz-demucs-valentini-28spk/source.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://huggingface.co/shahules786/mayavoz-demucs-valentini-28spk