Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Reed Wanderman-Milne's avatar
    Remove dynamic_loss_scale argument to define_performance. · 3803472a
    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
    3803472a
_performance.py 11.3 KB