Я не уверен, как получить время обучения для каждой эпохи. В настоящее время у меня есть объект TPUEstimator, который я тренирую с estimator.train () , но я просто возвращаю глобальные шаги и потери без времени:
Я пытаюсь повторить вывод, найденный здесь: https://colab.research.google.com/github/tensorflow/tpu/blob/master/tools/colab/mnist_tpuestimator.ipynb
Я чувствую, что, возможно, мне нужно настроить ведение журналавывод из Tensorflow. Однако я не уверен, как это сделать.