Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Reed Wanderman-Milne's avatar
    Remove dynamic_loss_scale argument to define_performance. · 20ed8cf0
    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
    20ed8cf0
resnet_cifar_main.py 10.1 KB