Merge pull request tensorflow#1565 from derekjchow/master

Reduce batchsize from 32->24 for SSD configs.
envin3 · Jun 15, 2017 · 6aa8eb4 · 6aa8eb4
2 parents df05d72 + 4b31d8f
commit 6aa8eb4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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: {