ptb_lm bug fix (#4448)

test=develop

ptb_lm bug fix (#4448)
test=develop
39cf4a2e · songyouwei · GitHub · 04b036cd · 39cf4a2e
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

dygraph/ptb_lm/ptb_dy.py dygraph/ptb_lm/ptb_dy.py +2 -2

未找到文件。
--- a/dygraph/ptb_lm/ptb_dy.py
+++ b/dygraph/ptb_lm/ptb_dy.py
@@ -394,8 +394,8 @@ def train_ptb_lm():
                dy_loss, last_hidden, last_cell = ptb_model(x, y, init_hidden,
                                                            init_cell)
-                init_hidden = last_hidden
+                init_hidden = last_hidden.detach()
-                init_cell = last_cell
+                init_cell = last_cell.detach()
                out_loss = dy_loss.numpy()
                dy_loss.backward()