如何使用tf.Session()。reset()?

时间:2019-05-13 05:11:14

标签: python python-3.x tensorflow

如何使用tf.Session().reset()?它和tf.Session().close()之间有什么关系?

with tf.Session() as sess:
   sess.run(something)
   """what is the actual syntax to use sess.reset() and how does it compare to sess.close()"""
   sess.reset()

1 个答案:

答案 0 :(得分:0)

sess.close()用于释放会话关联/分配的所有资源。例如,释放GPU内存和RAM。

sess.reset()做类似的工作,但是是针对分布式环境而定制的。应该在分布式会话上调用它,以停止分布式工作负载中的所有作业。

Source