Update i6_models/parts/frontend/generic_frontend.py

Co-authored-by: SimBe195 <[email protected]>
rwth-i6 · Nov 10, 2023 · 052a4a0 · 052a4a0
1 parent ec3356f
commit 052a4a0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/i6_models/parts/frontend/generic_frontend.py b/i6_models/parts/frontend/generic_frontend.py
@@ -199,8 +199,7 @@ def forward(self, tensor: torch.Tensor, sequence_mask: torch.Tensor) -> Tuple[to
         # and add a dim
         tensor = tensor[:, None, :, :]  # [B,C=1,T,F]
 
-        for i in range(len(self.cfg.layer_ordering)):
-            layer = self.frontend_layers[i]
+        for layer in self.frontend_layers:
             tensor = layer(tensor)
 
             if isinstance(layer, nn.Conv2d) or isinstance(layer, nn.MaxPool2d):