Keep only the last checkpoint around. And use cPikcle (pickle didn't help on the cloud).
Keep only the last checkpoint around. And use cPikcle (pickle didn't help on the cloud).

PiperOrigin-RevId: 250354246

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor