• Reed Wanderman-Milne's avatar
    Use nonexperimental mixed precision API. · 4334a892
    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
    4334a892
base_model.py 4.35 KB