subakany
commited on
Commit
·
8b20466
1
Parent(s):
9a645d8
update hyperparams
Browse files- hyperparams.yaml +0 -44
hyperparams.yaml
CHANGED
@@ -110,47 +110,6 @@ dataloader_opts:
|
|
110 |
|
111 |
|
112 |
# Specifying the network
|
113 |
-
Encoder: &id003 !new:speechbrain.lobes.models.dual_path.Encoder
|
114 |
-
kernel_size: 16
|
115 |
-
out_channels: 256
|
116 |
-
|
117 |
-
|
118 |
-
SBtfintra: &id001 !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
119 |
-
num_layers: 8
|
120 |
-
d_model: 256
|
121 |
-
nhead: 8
|
122 |
-
d_ffn: 1024
|
123 |
-
dropout: 0
|
124 |
-
use_positional_encoding: true
|
125 |
-
norm_before: true
|
126 |
-
|
127 |
-
SBtfinter: &id002 !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
128 |
-
num_layers: 8
|
129 |
-
d_model: 256
|
130 |
-
nhead: 8
|
131 |
-
d_ffn: 1024
|
132 |
-
dropout: 0
|
133 |
-
use_positional_encoding: true
|
134 |
-
norm_before: true
|
135 |
-
|
136 |
-
MaskNet: &id005 !new:speechbrain.lobes.models.dual_path.Dual_Path_Model
|
137 |
-
num_spks: 2
|
138 |
-
in_channels: 256
|
139 |
-
out_channels: 256
|
140 |
-
num_layers: 2
|
141 |
-
K: 250
|
142 |
-
intra_model: *id001
|
143 |
-
inter_model: *id002
|
144 |
-
norm: ln
|
145 |
-
linear_layer_after_inter_intra: false
|
146 |
-
skip_around_intra: true
|
147 |
-
|
148 |
-
Decoder: &id004 !new:speechbrain.lobes.models.dual_path.Decoder
|
149 |
-
in_channels: 256
|
150 |
-
out_channels: 1
|
151 |
-
kernel_size: 16
|
152 |
-
stride: 8
|
153 |
-
bias: false
|
154 |
|
155 |
snrmin: 0
|
156 |
snrmax: 10
|
@@ -259,9 +218,6 @@ epoch_counter: &id008 !new:speechbrain.utils.epoch_loop.EpochCounter
|
|
259 |
limit: 200
|
260 |
|
261 |
modules:
|
262 |
-
encoder: *id003
|
263 |
-
decoder: *id004
|
264 |
-
masknet: *id005
|
265 |
encoder: *id006
|
266 |
encoder_out: *id007
|
267 |
checkpointer: !new:speechbrain.utils.checkpoints.Checkpointer
|
|
|
110 |
|
111 |
|
112 |
# Specifying the network
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
113 |
|
114 |
snrmin: 0
|
115 |
snrmax: 10
|
|
|
218 |
limit: 200
|
219 |
|
220 |
modules:
|
|
|
|
|
|
|
221 |
encoder: *id006
|
222 |
encoder_out: *id007
|
223 |
checkpointer: !new:speechbrain.utils.checkpoints.Checkpointer
|