We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a6e044 commit 36f4de5Copy full SHA for 36f4de5
i6_models/decoder/attention.py
@@ -96,7 +96,6 @@ class AttentionLSTMDecoderV1(nn.Module):
96
def __init__(self, cfg: AttentionLSTMDecoderV1Config):
97
super().__init__()
98
99
- print(cfg.vocab_size)
100
self.target_embed = nn.Embedding(num_embeddings=cfg.vocab_size, embedding_dim=cfg.target_embed_dim)
101
self.target_embed_dropout = nn.Dropout(cfg.target_embed_dropout)
102
0 commit comments