Skip to content

Commit

Permalink
fix docs
Browse files Browse the repository at this point in the history
  • Loading branch information
mmz33 committed Nov 8, 2021
1 parent 5dc151a commit 6989cea
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nn/conformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ def __init__(
:param att_dropout: attention dropout value
:param out_dim: the output feature dimension
:param num_heads: the number of attention heads
:param batch_norm_opts: batch norm options
:param batch_norm_opts: passed to :class:`nn.BatchNorm`
"""
super().__init__()

Expand Down Expand Up @@ -181,7 +181,7 @@ class ConformerEncoder(nn.Module):

def __init__(self, encoder_layer: ConformerEncoderLayer, num_layers: int):
"""
:param encoder_layer: an instance of `class:ConformerEncoderLayer`
:param encoder_layer: an instance of :class:`ConformerEncoderLayer`
:param num_layers: the number of encoder layers
"""
super().__init__()
Expand Down

0 comments on commit 6989cea

Please sign in to comment.