swc2 commited on
Commit
6902860
·
verified ·
1 Parent(s): e20e413

upload base model

Browse files
DPRNN_TSE/LibriMix_Clean/Origin_mixing/best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f69b161dec0e1db3bacb25aafb868cacf51f80041aeb76bf5800296a61fead98
3
+ size 16192911
DPRNN_TSE/LibriMix_Clean/Origin_mixing/conf.yml ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data:
2
+ dm: false
3
+ spk_aug: false
4
+ n_src: 2
5
+ sample_rate: 8000
6
+ segment: 3
7
+ segment_aux: 3
8
+ spk_list: data/lib_2mix_extr_tr.spk
9
+ task: sep_clean
10
+ train_dir: data/wav8k/min/train-100
11
+ valid_dir: data/wav8k/min/dev
12
+ filterbank:
13
+ kernel_size: 2
14
+ n_filters: 64
15
+ stride: 1
16
+ main_args:
17
+ exp_dir: exp/tmp
18
+ help: null
19
+ masknet:
20
+ bidirectional: true
21
+ bn_chan: 128
22
+ chunk_size: 250
23
+ dropout: 0
24
+ hid_size: 128
25
+ hop_size: 125
26
+ in_chan: 64
27
+ mask_act: sigmoid
28
+ n_repeats: 6
29
+ num_spks: 291
30
+ out_chan: 64
31
+ optim:
32
+ lr: 0.001
33
+ optimizer: adam
34
+ weight_decay: 1.0e-05
35
+ positional arguments: {}
36
+ training:
37
+ batch_size: 3
38
+ early_stop: true
39
+ epochs: 200
40
+ gradient_clipping: 5
41
+ half_lr: true
42
+ last_checkpoint_path: null
43
+ num_workers: 4
44
+ reduce_patience: 5
DPRNN_TSE/LibriMix_Clean/Origin_mixing/out_best/all_metrics.csv ADDED
The diff for this file is too large to render. See raw diff
 
DPRNN_TSE/LibriMix_Clean/Origin_mixing/out_best/final_metrics.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "si_sdr": 13.695551985853662,
3
+ "si_sdr_imp": 13.695037737379938,
4
+ "sdr": 14.688920524650753,
5
+ "sdr_imp": 14.509687440776737,
6
+ "sar": 14.688920524650753,
7
+ "sar_imp": 14.509687440776737,
8
+ "stoi": 0.9081669595736415,
9
+ "stoi_imp": 0.19522648200343337
10
+ }
DPRNN_TSE/WSJ0-2mix/Origin_mixing/best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c5cd6d96ce1147d0bbaec30163e4be3dbbd55da8a3b2c83ae74754cb4579bcc
3
+ size 16094863
DPRNN_TSE/WSJ0-2mix/Origin_mixing/conf.yml ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data:
2
+ dm: false
3
+ spk_aug: false
4
+ n_src: 2
5
+ sample_rate: 8000
6
+ segment: 3
7
+ segment_aux: 3
8
+ spk_list: data/spk.scp
9
+ task: sep_clean
10
+ test_spk_list: data/test.spk
11
+ train_dir: data/wav8k/min/train
12
+ valid_dir: data/wav8k/min/dev
13
+ filterbank:
14
+ kernel_size: 2
15
+ n_filters: 64
16
+ stride: 1
17
+ main_args:
18
+ exp_dir: exp/tmp
19
+ help: null
20
+ masknet:
21
+ bidirectional: true
22
+ bn_chan: 128
23
+ chunk_size: 250
24
+ dropout: 0
25
+ hid_size: 128
26
+ hop_size: 125
27
+ in_chan: 64
28
+ mask_act: sigmoid
29
+ n_repeats: 6
30
+ num_spks: 101
31
+ out_chan: 64
32
+ optim:
33
+ lr: 0.001
34
+ optimizer: adam
35
+ weight_decay: 1.0e-05
36
+ positional arguments: {}
37
+ training:
38
+ batch_size: 3
39
+ early_stop: true
40
+ epochs: 200
41
+ gradient_clipping: 5
42
+ half_lr: true
43
+ last_checkpoint_path: null
44
+ num_workers: 4
45
+ reduce_patience: 5
DPRNN_TSE/WSJ0-2mix/Origin_mixing/out_best/all_metrics.csv ADDED
The diff for this file is too large to render. See raw diff
 
DPRNN_TSE/WSJ0-2mix/Origin_mixing/out_best/final_metrics.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "si_sdr": 17.71755104892538,
3
+ "si_sdr_imp": 17.718702475143907,
4
+ "stoi": 0.9448262607904043,
5
+ "stoi_imp": 0.2067801980424982
6
+ }