Frontends: Add tensor info to assert (#43)

* add tensor info to assert

---------

Co-authored-by: Albert Zeyer <[email protected]>

Author: Atticus1806

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]
 

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]