Skip to content

Commit 261e60c

Browse files
committed
added a docstring to forward_internal
Signed-off-by: Rauf <rnasretdinov@nvidia.com>
1 parent 821e3b8 commit 261e60c

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

nemo/collections/audio/parts/submodules/conformer_unet.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,15 @@ def forward_internal(
104104
cache_last_channel_len=None,
105105
bypass_pre_encode=None,
106106
):
107+
"""
108+
Forward pass for the ConformerEncoderUNet model with U-Net-style skip connections.
109+
110+
This method processes the input audio signal through the Conformer layers with optional
111+
caching and U-Net-style skip connections.
112+
113+
Main Changes Compared to Original Conformer:
114+
- Incorporates U-Net-style skip connections between encoder layers.
115+
"""
107116
if length is None:
108117
length = audio_signal.new_full(
109118
(audio_signal.size(0),), audio_signal.size(-1), dtype=torch.int64, device=audio_signal.device

0 commit comments

Comments
 (0)