Frontends: Add tensor info to assert (#43)

* add tensor info to assert --------- Co-authored-by: Albert Zeyer <[email protected]>
rwth-i6 · Dec 8, 2023 · 087d485 · 087d485
1 parent 933c6c1
commit 087d485
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/i6_models/parts/frontend/generic_frontend.py b/i6_models/parts/frontend/generic_frontend.py
@@ -195,7 +195,7 @@ def __init__(self, model_cfg: GenericFrontendV1Config):
             )
 
     def forward(self, tensor: torch.Tensor, sequence_mask: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
-        assert tensor.shape[-1] == self.cfg.in_features
+        assert tensor.shape[-1] == self.cfg.in_features, f"shape {tensor.shape} vs in features {self.cfg.in_features}"
         # and add a dim
         tensor = tensor[:, None, :, :]  # [B,C=1,T,F]
 

diff --git a/i6_models/parts/frontend/vgg_act.py b/i6_models/parts/frontend/vgg_act.py
@@ -158,7 +158,7 @@ def forward(self, tensor: torch.Tensor, sequence_mask: torch.Tensor) -> Tuple[to
         :param sequence_mask: the sequence mask for the tensor
         :return: torch.Tensor of shape [B,T",F'] and the shape of the sequence mask
         """
-        assert tensor.shape[-1] == self.cfg.in_features
+        assert tensor.shape[-1] == self.cfg.in_features, f"shape {tensor.shape} vs in features {self.cfg.in_features}"
         # and add a dim
         tensor = tensor[:, None, :, :]  # [B,C=1,T,F]