fix param doc (#46)

rwth-i6 · Mar 27, 2024 · 3c91736 · 3c91736
1 parent a15dad4
commit 3c91736
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/i6_models/assemblies/conformer/conformer_v1.py b/i6_models/assemblies/conformer/conformer_v1.py
@@ -52,7 +52,7 @@ def __init__(self, cfg: ConformerBlockV1Config):
     def forward(self, x: torch.Tensor, /, sequence_mask: torch.Tensor) -> torch.Tensor:
         """
         :param x: input tensor of shape [B, T, F]
-        :param sequence_mask: mask tensor where 0 defines positions within the sequence and 1 outside, shape: [B, T]
+        :param sequence_mask: mask tensor where 1 defines positions within the sequence and 0 outside, shape: [B, T]
         :return: torch.Tensor of shape [B, T, F]
         """
         x = 0.5 * self.ff1(x) + x  #  [B, T, F]
@@ -98,7 +98,7 @@ def __init__(self, cfg: ConformerEncoderV1Config):
     def forward(self, data_tensor: torch.Tensor, sequence_mask: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
         """
         :param data_tensor: input tensor of shape [B, T', F]
-        :param sequence_mask: mask tensor where 0 defines positions within the sequence and 1 outside, shape: [B, T']
+        :param sequence_mask: mask tensor where 1 defines positions within the sequence and 0 outside, shape: [B, T']
         :return: (output, out_seq_mask)
             where output is torch.Tensor of shape [B, T, F'],
             out_seq_mask is a torch.Tensor of shape [B, T]