Change LossScaleOptimizer checkpoint format.
Now the format is identical to as if a LossScaleOptimzier is not used, except that the loss scale is saved with a LossScaleOptimizer. This allows saving checkpoints with a LossScaleOptimizer and restoring without a LossScaleOptimizer, and vice versa. Checkpoints with LossScaleOptimizers created in older versions of TensorFlow can still be loaded. New checkpoints saved will use the new format. PiperOrigin-RevId: 306511555 Change-Id: Ie316ab8c4fbfec7babd6f7803d337799d0ff10a5
Showing
文件已添加
文件已添加
文件已添加
文件已添加
文件已添加
文件已添加
想要评论请 注册 或 登录