如何从TensorFlow测量sess.run（）的时间？

Question

如何衡量在会话中执行给定任务的执行时间？

sess.run(variables)

Answer 1

您可以使用python time包，正如Tensorflow作者在AlexNet Benchmark中所做的那样

start_time = time.time()
_ = session.run(variables)
duration = time.time() - start_time
print(duration)