如何使用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()
答案 0 :(得分:0)
sess.close()
用于释放会话关联/分配的所有资源。例如,释放GPU内存和RAM。
sess.reset()
做类似的工作,但是是针对分布式环境而定制的。应该在分布式会话上调用它,以停止分布式工作负载中的所有作业。