- 13 Jul, 2018 1 commit
-
-
Zhichao Lu authored
204489224 by Zhichao Lu: Modify ssd mobilenet v1 fpn config to be a bit more tolerant to OOM failure by bumping down the batch size to 64 and doubling the number of iterations to 25k. It now converges in 2.5 hours. -- 204488942 by Zhichao Lu: Internal change 204480631 by Zhichao Lu: This CL makes sure that num_steps parameter are not updated to 0 if num_steps field is not mentioned in config. The default behavior for number of steps parameter for training is infinite (train forever). The default value num_steps in train.proto is 0 (for training indefinitely). However the estimator/training function expects the num_steps to be set to None to train indefinitely. -- 204437217 by Zhichao Lu: Create a Docker image to support TensorFlow Lite / Object Detection blog post. -- 204317570 by Zhichao Lu: Internal change PiperOrigin-RevId: 204489224
-