• Reed Wanderman-Milne's avatar
    Use nonexperimental mixed precision API. · 19d18c00
    Reed Wanderman-Milne authored
    This replaces symbols in tf.keras.mixed_precision.experimental with the corresponding nonexperimental symbols. In some cases, passing a Policy is replaced with passing a policy name for conciseness.
    
    Additionally, for the Shakespeare model, the loss_scale flag is removed, since supporting it with the nonexperimental API is slightly more verbose and it is recommended users use the default loss scale.
    
    PiperOrigin-RevId: 368123944
    19d18c00
model_lib_v2.py 47.5 KB