如何重定向TFDBG转储目录



默认情况下, tfdbg将保存的张量转储到 /tmp,但是在我的情况下, /tmp安装在/root中,/root只有几个G的空间,运行示例调试不是问题,但是当调试大型网络,其中一项运行将生成超过10 g内存的张量,它会提示空间不足。

在https://github.com/tensorflow/tensorflow/16867

中回答

constructors localclidebugwrappersession和localclidebughook都有一个称为dump_root的关键字参数,该参数允许您指定在何处转储调试数据。

请参阅:

https://www.tensorflow.org/api_docs/python/tfdbg/localclidebugwwrappersessessuppersession# in Init

https://www.tensorflow.org/api_docs/python/tfdbg/localclidebughook# init

最新更新