Skip to content

Commit e646893

Browse files
author
chengduo
authored
Merge pull request PaddlePaddle#77 from chengduoZH/fix_PaddingRNN
Fix bug of PaddingRnn/lm_model.py
2 parents 18f532b + a0535fa commit e646893

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

PaddingRNN/lstm_paddle/lm_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -246,13 +246,13 @@ def encoder_static(input_embedding, len=3, init_hidden=None,
246246
init_cell = layers.data(name="init_cell", shape=[num_layers, batch_size, hidden_size],
247247
dtype='float32', append_batch_size=False)
248248

249+
init_cell.persistable = True
250+
init_hidden.persistable = True
251+
249252
init_hidden = layers.reshape(
250253
init_hidden, shape=[num_layers, -1, hidden_size])
251254
init_cell = layers.reshape(init_cell, shape=[num_layers, -1, hidden_size])
252255

253-
init_cell.persistable = True
254-
init_hidden.persistable = True
255-
256256
x_emb = layers.embedding(
257257
input=x,
258258
size=[vocab_size, hidden_size],

0 commit comments

Comments
 (0)