Checkpoint commit was successful to gs://prpl-ckpt-us-central2-944301850228/lj8040/5_pp_pen_open_book_prpl/checkpoints/5_pp_pen_open_book_prpl/5_pp_pen_open_book_prpl/19999/train_state