cemsubakan
commited on
Commit
•
d846584
1
Parent(s):
96f94b4
Update hyperparams.yaml
Browse files- hyperparams.yaml +8 -8
hyperparams.yaml
CHANGED
@@ -15,11 +15,11 @@ kernel_size: 16
|
|
15 |
kernel_stride: 8
|
16 |
|
17 |
# Specifying the network
|
18 |
-
Encoder:
|
19 |
kernel_size: 16
|
20 |
out_channels: 256
|
21 |
|
22 |
-
SBtfintra:
|
23 |
num_layers: 8
|
24 |
d_model: 256
|
25 |
nhead: 8
|
@@ -28,7 +28,7 @@ SBtfintra: &id001 !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
|
28 |
use_positional_encoding: true
|
29 |
norm_before: true
|
30 |
|
31 |
-
SBtfinter:
|
32 |
num_layers: 8
|
33 |
d_model: 256
|
34 |
nhead: 8
|
@@ -37,7 +37,7 @@ SBtfinter: &id002 !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
|
37 |
use_positional_encoding: true
|
38 |
norm_before: true
|
39 |
|
40 |
-
MaskNet:
|
41 |
num_spks: 3
|
42 |
in_channels: 256
|
43 |
out_channels: 256
|
@@ -49,7 +49,7 @@ MaskNet: &id005 !new:speechbrain.lobes.models.dual_path.Dual_Path_Model
|
|
49 |
linear_layer_after_inter_intra: false
|
50 |
skip_around_intra: true
|
51 |
|
52 |
-
Decoder:
|
53 |
in_channels: 256
|
54 |
out_channels: 1
|
55 |
kernel_size: 16
|
@@ -57,9 +57,9 @@ Decoder: &id004 !new:speechbrain.lobes.models.dual_path.Decoder
|
|
57 |
bias: false
|
58 |
|
59 |
modules:
|
60 |
-
encoder:
|
61 |
-
decoder:
|
62 |
-
masknet:
|
63 |
|
64 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
65 |
loadables:
|
|
|
15 |
kernel_stride: 8
|
16 |
|
17 |
# Specifying the network
|
18 |
+
Encoder: !new:speechbrain.lobes.models.dual_path.Encoder
|
19 |
kernel_size: 16
|
20 |
out_channels: 256
|
21 |
|
22 |
+
SBtfintra: !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
23 |
num_layers: 8
|
24 |
d_model: 256
|
25 |
nhead: 8
|
|
|
28 |
use_positional_encoding: true
|
29 |
norm_before: true
|
30 |
|
31 |
+
SBtfinter: !new:speechbrain.lobes.models.dual_path.SBTransformerBlock
|
32 |
num_layers: 8
|
33 |
d_model: 256
|
34 |
nhead: 8
|
|
|
37 |
use_positional_encoding: true
|
38 |
norm_before: true
|
39 |
|
40 |
+
MaskNet: !new:speechbrain.lobes.models.dual_path.Dual_Path_Model
|
41 |
num_spks: 3
|
42 |
in_channels: 256
|
43 |
out_channels: 256
|
|
|
49 |
linear_layer_after_inter_intra: false
|
50 |
skip_around_intra: true
|
51 |
|
52 |
+
Decoder: !new:speechbrain.lobes.models.dual_path.Decoder
|
53 |
in_channels: 256
|
54 |
out_channels: 1
|
55 |
kernel_size: 16
|
|
|
57 |
bias: false
|
58 |
|
59 |
modules:
|
60 |
+
encoder: !ref <Encoder>
|
61 |
+
decoder: !ref <Decoder>
|
62 |
+
masknet: !ref <MaskNet>
|
63 |
|
64 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
65 |
loadables:
|