From 40e28de4acf962dad39dcd34f906c02289741d19 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Tue, 16 Apr 2019 16:06:52 +0800 Subject: [PATCH] Add train in mnist ut test=release/1.4 --- python/paddle/fluid/tests/unittests/test_imperative_mnist.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/tests/unittests/test_imperative_mnist.py b/python/paddle/fluid/tests/unittests/test_imperative_mnist.py index 5ab01839fbc..6aae2c0507d 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_mnist.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_mnist.py @@ -117,6 +117,7 @@ class TestImperativeMnist(unittest.TestCase): train_reader = paddle.batch( paddle.dataset.mnist.train(), batch_size=128, drop_last=True) + mnist.train() dy_param_init_value = {} for epoch in range(epoch_num): for batch_id, data in enumerate(train_reader()): -- GitLab