Reduce batchsize from 32->24 for SSD configs.

derekjchow-google · derekjchow-google · commit 4b31d8fee787 · 2017-06-15T10:26:59.000-07:00
diff --git a/object_detection/samples/configs/ssd_inception_v2_pets.config b/object_detection/samples/configs/ssd_inception_v2_pets.config
@@ -134,7 +134,7 @@ model {
 }
 
 train_config: {
-  batch_size: 32
+  batch_size: 24
   optimizer {
     rms_prop_optimizer: {
       learning_rate: {
diff --git a/object_detection/samples/configs/ssd_mobilenet_v1_pets.config b/object_detection/samples/configs/ssd_mobilenet_v1_pets.config
@@ -140,7 +140,7 @@ model {
 }
 
 train_config: {
-  batch_size: 32
+  batch_size: 24
   optimizer {
     rms_prop_optimizer: {
       learning_rate: {
