Merge branch 'master' of upstream into fluid_mm_dnn

805094591 · Sep 14, 2018 · ea74f4c · ea74f4c
2 parents a14d526 + 4dbe7f7
commit ea74f4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/simnet/train/paddle/paddle_simnet.py b/tools/simnet/train/paddle/paddle_simnet.py
@@ -161,7 +161,7 @@ def predict(conf_dict):
                     place=place, feed_list=feed_var_names, program=program)
                 reader = data_reader.get_reader(conf_dict, True, samples_file)
             # Get batch data iterator
-            batch_data = paddle.batch(reader, conf_dict["batch_size"])
+            batch_data = paddle.batch(reader, conf_dict["batch_size"], drop_last=False)
             logging.info("start test process ...")
             for iter, data in enumerate(batch_data()):
                 output = executor.run(program, feed=feeder.feed(