• Reed Wanderman-Milne's avatar
    Remove dynamic_loss_scale argument to define_performance. · e353e4e5
    Reed Wanderman-Milne authored
    All models which support loss scaling support dynamic loss scaling, so the argument has no purpose. It used to be that some models scaled the loss manually instead of using a LossScaleOptimizer, and so did not support dynamic loss scaling.
    
    PiperOrigin-RevId: 367719521
    e353e4e5
flags_test.py 5.18 KB