From c9598c0901e18745e38738f29107198cf7368203 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Fri, 29 Mar 2024 21:34:20 +0000 Subject: [PATCH] formatting --- .../librispeech_conformer/librispeech_pytorch/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py index 90a12b779..502cb093e 100644 --- a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py +++ b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py @@ -208,7 +208,7 @@ def __init__(self, config: ConformerConfig): config.feed_forward_residual_dropout_rate) self.dropout2 = nn.Dropout( p=feed_forward_residual_dropout_rate, inplace=True) - + def forward(self, inputs, padding_mask): inputs = self.ln(inputs) inputs = self.linear1(inputs)