[Dy2stat] Add Ptb Lm Test for V2 APIs (#27461)

As the title
4 years ago · 28f83f0519
parent b77d9f268e
commit 28f83f0519
2 changed files with 315 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py
@ -279,7 +279,7 @@ def train(place):
                               speed))
                        avg_batch_time = time.time()

-            return out_loss, last_hidden.numpy(), last_cell.numpy()
+        return out_loss, last_hidden.numpy(), last_cell.numpy()


 def train_dygraph(place):
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm_v2.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm_v2.py