Add model saving in Resnet50 code.
In addition, 1. Optimized the weights saving logic in Keras for TPU. 2. Fixed the bug in optimizer for TPU mirrored variable. PiperOrigin-RevId: 227704010
Loading
Please sign in to comment
In addition, 1. Optimized the weights saving logic in Keras for TPU. 2. Fixed the bug in optimizer for TPU mirrored variable. PiperOrigin-RevId: 227704010