Параметры отладки Tensorflow

Question

Я использую модель, которая потребляет слишком много памяти.Есть ли у tenorflow опция, с помощью которой я могу узнать, какие операции занимают сколько памяти?

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