Skip to content

Commit ea4354c

Browse files
somehow-workingOle
andauthored
fixed typos (#45)
Co-authored-by: Ole <[email protected]>
1 parent 087d485 commit ea4354c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

i6_models/assemblies/e_branchformer/e_branchformer_v1.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,15 +66,16 @@ def forward(self, x: torch.Tensor, /, sequence_mask: torch.Tensor) -> torch.Tens
6666
:param sequence_mask: mask tensor where 0 defines positions within the sequence and 1 outside, shape: [B, T]
6767
:return: torch.Tensor of shape [B, T, F]
6868
"""
69-
x = 0.5 * self.ff1(x) + x # [B, T, F]
69+
x = 0.5 * self.ff_1(x) + x # [B, T, F]
7070
x_1 = self.mhsa(x, sequence_mask) # [B, T, F]
7171
x_2 = self.cgmlp(x) # [B, T, F]
7272
x = self.merger(x_1, x_2) + x # [B, T, F]
73-
x = 0.5 * self.ff2(x) + x # [B, T, F]
73+
x = 0.5 * self.ff_2(x) + x # [B, T, F]
7474
x = self.final_layer_norm(x) # [B, T, F]
7575
return x
7676

7777

78+
@dataclass
7879
class EbranchformerEncoderV1Config(ModelConfiguration):
7980
"""
8081
Attributes:

0 commit comments

Comments
 (0)