Skip to content

Commit 42d0e42

Browse files
Fixed LSTM model
1 parent 5e04f13 commit 42d0e42

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

boudams/model/lstm.py

-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ def __init__(self, input_dim, emb_dim, hid_dim, n_layers, dropout):
2424
def forward(self, src, src_len):
2525

2626
# src = [src sent len, batch size]
27-
2827
embedded = self.dropout(self.embedding(src))
2928

3029
# embedded = [src sent len, batch size, emb dim]

configs/bfmmss-lstm.json

+2-3
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,10 @@
22
"name": "bfmmss-lstm",
33
"max_sentence_size": 150,
44
"network": {
5-
"emb_enc_dim": 512,
6-
"enc_kernel_size": 5,
5+
"emb_enc_dim": 256,
76
"enc_dropout": 0.5,
87
"enc_hid_dim": 512,
9-
"enc_n_layers": 10
8+
"enc_n_layers": 2
109
},
1110
"model": "linear-lstm",
1211
"batch_size": 32,

0 commit comments

Comments
 (0)