Wangyou Zhang commited on
Commit
b11823b
1 Parent(s): 133b80f

Add model files

Browse files
exp/enh_stats_16k/train/feats_stats.npz ADDED
Binary file (742 Bytes). View file
exp/enh_train_enh_conv_tasnet_raw/5epoch.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:afedfa8ff65bf256b02e390301ee75ba67dc5d7d1f83f48a76aa2b9610c44c86
3
+ size 34707333
exp/enh_train_enh_conv_tasnet_raw/RESULTS.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <!-- Generated by ./scripts/utils/show_enh_score.sh -->
2
+ # RESULTS
3
+ ## Environments
4
+ - date: `Thu Jan 28 22:56:03 CST 2021`
5
+ - python version: `3.6.3 |Anaconda, Inc.| (default, Nov 20 2017, 20:41:42) [GCC 7.2.0]`
6
+ - espnet version: `espnet 0.9.7`
7
+ - pytorch version: `pytorch 1.6.0`
8
+ - Git hash: `eddccfa7a296fcf8fd744200d7e8113f9a3c7e69`
9
+ - Commit date: `Thu Jan 21 21:42:24 2021 +0800`
10
+
11
+
12
+ ## enh_train_enh_conv_tasnet_raw
13
+
14
+ config: conf/tuning/train_enh_conv_tasnet.yaml
15
+
16
+ |dataset|PESQ|STOI|SAR|SDR|SIR|SI_SNR|
17
+ |---|---|---|---|---|---|---|
18
+ |enhanced_dt05_simu_isolated_1ch_track|2.63359|0.909127|11.3937|11.3937|0|10.1792|
19
+ |enhanced_et05_simu_isolated_1ch_track|2.58602|0.891826|12.5193|12.5193|0|11.5768|
20
+
exp/enh_train_enh_conv_tasnet_raw/config.yaml ADDED
@@ -0,0 +1,137 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config: conf/tuning/train_enh_conv_tasnet.yaml
2
+ print_config: false
3
+ log_level: INFO
4
+ dry_run: false
5
+ iterator_type: chunk
6
+ output_dir: exp/enh_train_enh_conv_tasnet_raw
7
+ ngpu: 1
8
+ seed: 0
9
+ num_workers: 4
10
+ num_att_plot: 3
11
+ dist_backend: nccl
12
+ dist_init_method: env://
13
+ dist_world_size: 2
14
+ dist_rank: 0
15
+ local_rank: 0
16
+ dist_master_addr: localhost
17
+ dist_master_port: 57680
18
+ dist_launcher: null
19
+ multiprocessing_distributed: true
20
+ cudnn_enabled: true
21
+ cudnn_benchmark: false
22
+ cudnn_deterministic: true
23
+ collect_stats: false
24
+ write_collected_feats: false
25
+ max_epoch: 100
26
+ patience: 4
27
+ val_scheduler_criterion:
28
+ - valid
29
+ - loss
30
+ early_stopping_criterion:
31
+ - valid
32
+ - loss
33
+ - min
34
+ best_model_criterion:
35
+ - - valid
36
+ - si_snr
37
+ - max
38
+ - - valid
39
+ - loss
40
+ - min
41
+ keep_nbest_models: 1
42
+ grad_clip: 5.0
43
+ grad_clip_type: 2.0
44
+ grad_noise: false
45
+ accum_grad: 1
46
+ no_forward_run: false
47
+ resume: true
48
+ train_dtype: float32
49
+ use_amp: false
50
+ log_interval: null
51
+ unused_parameters: false
52
+ use_tensorboard: true
53
+ use_wandb: false
54
+ wandb_project: null
55
+ wandb_id: null
56
+ pretrain_path: null
57
+ init_param: []
58
+ freeze_param: []
59
+ num_iters_per_epoch: null
60
+ batch_size: 8
61
+ valid_batch_size: null
62
+ batch_bins: 1000000
63
+ valid_batch_bins: null
64
+ train_shape_file:
65
+ - exp/enh_stats_16k/train/speech_mix_shape
66
+ - exp/enh_stats_16k/train/speech_ref1_shape
67
+ valid_shape_file:
68
+ - exp/enh_stats_16k/valid/speech_mix_shape
69
+ - exp/enh_stats_16k/valid/speech_ref1_shape
70
+ batch_type: folded
71
+ valid_batch_type: null
72
+ fold_length:
73
+ - 80000
74
+ - 80000
75
+ sort_in_batch: descending
76
+ sort_batch: descending
77
+ multiple_iterator: false
78
+ chunk_length: 32000
79
+ chunk_shift_ratio: 0.5
80
+ num_cache_chunks: 1024
81
+ train_data_path_and_name_and_type:
82
+ - - dump/raw/tr05_simu_isolated_1ch_track/wav.scp
83
+ - speech_mix
84
+ - sound
85
+ - - dump/raw/tr05_simu_isolated_1ch_track/spk1.scp
86
+ - speech_ref1
87
+ - sound
88
+ valid_data_path_and_name_and_type:
89
+ - - dump/raw/dt05_simu_isolated_1ch_track/wav.scp
90
+ - speech_mix
91
+ - sound
92
+ - - dump/raw/dt05_simu_isolated_1ch_track/spk1.scp
93
+ - speech_ref1
94
+ - sound
95
+ allow_variable_data_keys: false
96
+ max_cache_size: 0.0
97
+ max_cache_fd: 32
98
+ valid_max_cache_size: null
99
+ optim: adam
100
+ optim_conf:
101
+ lr: 0.001
102
+ eps: 1.0e-08
103
+ weight_decay: 1.0e-05
104
+ scheduler: reducelronplateau
105
+ scheduler_conf:
106
+ mode: min
107
+ factor: 0.5
108
+ patience: 3
109
+ init: xavier_uniform
110
+ model_conf:
111
+ loss_type: si_snr
112
+ use_preprocessor: false
113
+ encoder: conv
114
+ encoder_conf:
115
+ channel: 256
116
+ kernel_size: 20
117
+ stride: 10
118
+ separator: tcn
119
+ separator_conf:
120
+ num_spk: 1
121
+ layer: 8
122
+ stack: 4
123
+ bottleneck_dim: 256
124
+ hidden_dim: 512
125
+ kernel: 3
126
+ causal: false
127
+ norm_type: gLN
128
+ nonlinear: relu
129
+ decoder: conv
130
+ decoder_conf:
131
+ channel: 256
132
+ kernel_size: 20
133
+ stride: 10
134
+ required:
135
+ - output_dir
136
+ version: 0.9.7
137
+ distributed: true
exp/enh_train_enh_conv_tasnet_raw/images/backward_time.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/forward_time.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/iter_time.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/loss.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/lr_0.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/optim_step_time.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/si_snr.png ADDED
exp/enh_train_enh_conv_tasnet_raw/images/train_time.png ADDED
exp/enh_train_enh_conv_tasnet_raw/valid.loss.best.pth ADDED
@@ -0,0 +1 @@
 
1
+ 5epoch.pth
meta.yaml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
1
+ espnet: 0.9.9
2
+ files:
3
+ model_file: exp/enh_train_enh_conv_tasnet_raw/5epoch.pth
4
+ python: "3.7.6 (default, Jan 8 2020, 19:59:22) \n[GCC 7.3.0]"
5
+ timestamp: 1622603546.534792
6
+ torch: 1.7.1
7
+ yaml_files:
8
+ train_config: exp/enh_train_enh_conv_tasnet_raw/config.yaml