• Reed Wanderman-Milne's avatar
    Use nonexperimental mixed precision API. · c0ac8d1c
    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
    c0ac8d1c
unet3d_benchmark.py 5.77 KB