Add train in mnist ut

velconia · velconia · commit 40e28de4acf9 · 2019-04-16T16:06:52.000+08:00
test=release/1.4
diff --git a/python/paddle/fluid/tests/unittests/test_imperative_mnist.py b/python/paddle/fluid/tests/unittests/test_imperative_mnist.py
@@ -117,6 +117,7 @@ def test_mnist_float32(self):
             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()):
