(self, sty_dim, d_model, nlayers, dropout=0.1)
| 136 | |
| 137 | class DurationEncoder(nn.Module): |
| 138 | def __init__(self, sty_dim, d_model, nlayers, dropout=0.1): |
| 139 | super().__init__() |
| 140 | self.lstms = nn.ModuleList() |
| 141 | for _ in range(nlayers): |
| 142 | self.lstms.append(nn.LSTM(d_model + sty_dim, d_model // 2, num_layers=1, batch_first=True, bidirectional=True, dropout=dropout)) |
| 143 | self.lstms.append(AdaLayerNorm(sty_dim, d_model)) |
| 144 | self.dropout = dropout |
| 145 | self.d_model = d_model |
| 146 | self.sty_dim = sty_dim |
| 147 | |
| 148 | def forward(self, x, style, text_lengths, m): |
| 149 | masks = m |
nothing calls this directly
no test coverage detected