如何从TensorFlow测量sess.run()的时间？

Question

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

sess.run(variables)

Answer 1

您可以使用python time包,正如Tensorflow作者在AlexNet基准测试中所做的那样

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