Merge pull request #695 from qingqing01/mobilenet_ssd

Fix the test reader for SSD.
PaddlePaddle · Mar 8, 2018 · df8060e · df8060e
2 parents 3483068 + 609f7aa
commit df8060e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fluid/object_detection/mobilenet_ssd.py b/fluid/object_detection/mobilenet_ssd.py
@@ -174,7 +174,7 @@ def train(train_file_list,
     train_reader = paddle.batch(
         reader.train(data_args, train_file_list), batch_size=batch_size)
     test_reader = paddle.batch(
-        reader.test(data_args, train_file_list), batch_size=batch_size)
+        reader.test(data_args, val_file_list), batch_size=batch_size)
     feeder = fluid.DataFeeder(
         place=place, feed_list=[image, gt_box, gt_label, difficult])
 
@@ -217,6 +217,6 @@ def train(train_file_list,
         train_file_list='./data/trainval.txt',
         val_file_list='./data/test.txt',
         data_args=data_args,
-        learning_rate=0.004,
+        learning_rate=0.001,
         batch_size=32,
         num_passes=300)