如何重定向tfdbg转储目录

时间:2018-02-08 14:13:16

标签: tensorflow

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

1 个答案:

答案 0 :(得分:0)

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

中回答

构造函数LocalCLIDebugWrapperSession和LocalCLIDebugHook都有一个名为dump_root的关键字参数,它允许您指定转储调试数据的位置。

请参阅:

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

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