如何使用TPUEstimator获得每个时期的训练时间？

Question

我不确定如何获取每个时期的训练时间。目前，我有一个使用 estimator.train（）训练的TPUEstimator对象，但我只是回到全局步骤而又不花时间：

我正在尝试复制此处找到的输出：https://colab.research.google.com/github/tensorflow/tpu/blob/master/tools/colab/mnist_tpuestimator.ipynb

我觉得也许我必须设置Tensorflow的日志输出。但是，我不确定该怎么做。