1. 11 Oct, 2021 2 commits
  2. 09 Oct, 2021 2 commits
  3. 05 Apr, 2021 2 commits
    • Reed Wanderman-Milne's avatar
      Use nonexperimental LSO API in base_task.py. · 2ca3440e
      Reed Wanderman-Milne authored
      This shouldn't break any official models, since I changed all LossScaleOptimizer isinstance checks to use the nonexperimental version (the experimental LSO subclasses the nonexperimental LSO, so changing isinstance checks in this way is always safe).
      
      PiperOrigin-RevId: 366891847
      2ca3440e
    • Reed Wanderman-Milne's avatar
      Use nonexperimental LSO API in base_task.py. · cc12499b
      Reed Wanderman-Milne authored
      This shouldn't break any official models, since I changed all LossScaleOptimizer isinstance checks to use the nonexperimental version (the experimental LSO subclasses the nonexperimental LSO, so changing isinstance checks in this way is always safe).
      
      PiperOrigin-RevId: 366891847
      cc12499b
  4. 25 Mar, 2021 1 commit