Skip to content

Commit 3913238

Browse files
committed
formatting
1 parent 7ca8365 commit 3913238

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

  • algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax

algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/models.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -616,7 +616,12 @@ def __call__(self,
616616
inputs, input_paddings, train)
617617

618618
inputs = inputs + \
619-
ConvolutionBlock(config)(inputs, input_paddings, train, update_batch_norm, use_running_average)
619+
ConvolutionBlock(config)(inputs,
620+
input_paddings,
621+
train,
622+
update_batch_norm,
623+
use_running_average
624+
)
620625

621626
inputs = inputs + 0.5 * FeedForwardModule(config=self.config)(
622627
inputs, padding_mask, train)

0 commit comments

Comments
 (0)