From d085f486f11dc869d5ac3125f826cb2f1933f4ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sat, 23 Sep 2023 20:16:14 +0800 Subject: [PATCH] Update models.py --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index dd9e0c0..1e103c7 100644 --- a/models.py +++ b/models.py @@ -340,7 +340,7 @@ class TextEncoder(nn.Module): self.language_emb = nn.Embedding(num_languages, hidden_channels) nn.init.normal_(self.language_emb.weight, 0.0, hidden_channels**-0.5) self.bert_proj = nn.Conv1d(1024, hidden_channels, 1) - self.ja_bert_proj = nn.Conv1d(768, hidden_channels, 1) + self.ja_bert_proj = nn.Conv1d(1024, hidden_channels, 1) self.encoder = attentions.Encoder( hidden_channels,