Yabin Li
2024-03-20 3958472fb2a9bbac5cb2a30e3fb11925c7b5d3d8
funasr/models/eend/encoder.py
@@ -7,7 +7,7 @@
class MultiHeadSelfAttention(nn.Module):
    def __init__(self, n_units, h=8, dropout_rate=0.1):
        super(MultiHeadSelfAttention, self).__init__()
        super().__init__()
        self.linearQ = nn.Linear(n_units, n_units)
        self.linearK = nn.Linear(n_units, n_units)
        self.linearV = nn.Linear(n_units, n_units)