夜雨飘零
2024-02-02 c2f174cd7811a7a11e6227ecb371887f97fd66d3
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)