Skip to content

GitLab

Commit 4b31d8fe authored by Derek Chow's avatar Derek Chow
Browse files

Reduce batchsize from 32->24 for SSD configs.

parent b0a13bfa
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
object_detection/samples/configs/ssd_inception_v2_pets.config
View file @ 4b31d8fe
...@@ -134,7 +134,7 @@ model { ...@@ -134,7 +134,7 @@ model {
} }
train_config: { train_config: {
batch_size: 32 batch_size: 24
optimizer { optimizer {
rms_prop_optimizer: { rms_prop_optimizer: {
learning_rate: { learning_rate: {
......
object_detection/samples/configs/ssd_mobilenet_v1_pets.config
View file @ 4b31d8fe
...@@ -140,7 +140,7 @@ model { ...@@ -140,7 +140,7 @@ model {
} }
train_config: { train_config: {
batch_size: 32 batch_size: 24
optimizer { optimizer {
rms_prop_optimizer: { rms_prop_optimizer: {
learning_rate: { learning_rate: {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment