added conformerencoder to all.

Signed-off-by: Vahid <vnoroozi@nvidia.com>
NVIDIA · Nov 3, 2020 · 302b9f3 · 302b9f3
1 parent 9c1a652
commit 302b9f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/parts/conformer_modules.py b/nemo/collections/asr/parts/conformer_modules.py
@@ -19,7 +19,7 @@
 from nemo.collections.asr.parts.activations import Swish
 from nemo.collections.asr.parts.multi_head_attention import MultiHeadAttention, RelPositionMultiHeadAttention
 
-__all__ = ['ConformerConvolution', 'ConformerFeedForward']
+__all__ = ['ConformerConvolution', 'ConformerFeedForward', 'ConformerEncoderBlock']
 
 
 class ConformerEncoderBlock(torch.nn.Module):