• 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...
    376dc8dd
README.md 24.5 KB