From 4b31d8fee78729aa2964c9201b0cd344eba6be8d Mon Sep 17 00:00:00 2001 From: Derek Chow Date: Thu, 15 Jun 2017 10:26:59 -0700 Subject: [PATCH] Reduce batchsize from 32->24 for SSD configs. --- object_detection/samples/configs/ssd_inception_v2_pets.config | 2 +- object_detection/samples/configs/ssd_mobilenet_v1_pets.config | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/object_detection/samples/configs/ssd_inception_v2_pets.config b/object_detection/samples/configs/ssd_inception_v2_pets.config index 801701ed55e..49bdf7e0643 100644 --- 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 index e8b0516f22f..c8d83ddb146 100644 --- 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: {