• Zhichao Lu's avatar
    Merged commit includes the following changes: · 85dd5fa4
    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
    85dd5fa4
Dockerfile 5.57 KB