• Mark Sandler's avatar
    Pulling internal changes to github (#3583) · 376dc8dd
    Mark Sandler authored
    * Internal change.
    
    PiperOrigin-RevId: 187042423
    
    * Internal change.
    
    PiperOrigin-RevId: 187072380
    
    * Opensource float and eight-bit fixed-point mobilenet_v1 training and eval scripts.
    
    PiperOrigin-RevId: 187106140
    
    * Initial check-in for Mobilenet V2
    
    PiperOrigin-RevId: 187213595
    
    * Allow configuring batch normalization decay and epsilon in MobileNet v1
    
    PiperOrigin-RevId: 187425294
    
    * Allow overriding NASNet model HParams.
    
    This is a change to the API that will allow users to pass in their own configs
    to the building functions, which should make these APIs much more customizable
    for end-user cases.
    
    This change removes the use_aux_head argument from the model construction
    functions, which is no longer necessary given that the use_aux_head option is
    configurable in the model config. For example, for the mobile ImageNet model,
    the auxiliary head can be disabled using:
    
    config = nasnet.mobile_imagenet_config()
    config.set_hparam('use_aux_head', 0)
    logits, endpoints = nasnet.build_nasnet_mobile(
        inputs, num_classes, config=config)
    PiperOrigin-RevId: 188617685
    
    * Automated g4 rollback of changelist 188617685
    
    PiperOrigin-RevId: 188619139
    
    * Removes spurious comment
    376dc8dd
inception_preprocessing.py 13.8 KB