Skip to content

Conversation

pkulzc
Copy link
Contributor

@pkulzc pkulzc commented Jul 13, 2018

  • Moving from slim-based training to Estimator-based training.

  • Support for RetinaNet, and a MobileNet adaptation of RetinaNet.

  • A novel SSD-based architecture called the Pooling Pyramid Network (PPN).

  • Releasing several TPU-compatible models. These can be found in the samples/configs/ directory with a comment in the pipeline configuration files indicating TPU compatibility.

  • Support for quantized training.

  • Updated documentation for new binaries, Cloud training, and Tensorflow Lite.

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
@pkulzc pkulzc requested a review from jch1 July 13, 2018 18:32
@pkulzc pkulzc requested a review from derekjchow as a code owner July 13, 2018 18:32
Copy link

@jch1 jch1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome, thanks Zhichao!!!

@jch1 jch1 merged commit 70b176a into tensorflow:master Jul 13, 2018
@austinmw
Copy link

Hi, is there a RetinaNet config file for the model zoo, or is this model ran differently?

@hitlk
Copy link

hitlk commented Jul 17, 2018

@austinmw
samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config is for RetinaNet.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants