upload base model
Browse files- DPRNN_TSE/LibriMix_Clean/Origin_mixing/best_model.pth +3 -0
- DPRNN_TSE/LibriMix_Clean/Origin_mixing/conf.yml +44 -0
- DPRNN_TSE/LibriMix_Clean/Origin_mixing/out_best/all_metrics.csv +0 -0
- DPRNN_TSE/LibriMix_Clean/Origin_mixing/out_best/final_metrics.json +10 -0
- DPRNN_TSE/WSJ0-2mix/Origin_mixing/best_model.pth +3 -0
- DPRNN_TSE/WSJ0-2mix/Origin_mixing/conf.yml +45 -0
- DPRNN_TSE/WSJ0-2mix/Origin_mixing/out_best/all_metrics.csv +0 -0
- DPRNN_TSE/WSJ0-2mix/Origin_mixing/out_best/final_metrics.json +6 -0
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 |
+
}
|