Reduce batchsize from 32->24 for SSD configs.

envin3 · Jun 15, 2017 · 4b31d8f · 4b31d8f
1 parent b0a13bf
commit 4b31d8f
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: {