Merge pull request #1090 from luotao1/fix_image_classification_error

fix image_classification error

fluid/image_classification/reader.py

@@ -160,5 +160,5 @@ def val(file_list=TEST_LIST):
     return _reader_creator(file_list, 'val', shuffle=False)
 
 
-def test(file_list):
+def test(file_list=TEST_LIST):
     return _reader_creator(file_list, 'test', shuffle=False)

fluid/image_classification/train.py

@@ -157,7 +157,8 @@ def train(args):
     test_reader = paddle.batch(reader.val(), batch_size=test_batch_size)
     feeder = fluid.DataFeeder(place=place, feed_list=[image, label])
 
-    train_exe = fluid.ParallelExecutor(use_cuda=True, loss_name=avg_cost.name)
+    train_exe = fluid.ParallelExecutor(
+        use_cuda=True if args.use_gpu else False, loss_name=avg_cost.name)
 
     fetch_list = [avg_cost.name, acc_top1.name, acc_top5.name]