Tensorflow调试选项

Question

我正在运行一个消耗太多内存的模型。 tensorflow是否有一个选项可以让我知道哪些操作正在消耗多少内存？

Answer 1

我建议使用Tensorflow Profiler;

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/profiler/g3doc/profile_memory.md