TensorFlow: How to measure how much GPU memory each tensor takes?

Question

I\'m currently implementing YOLO in TensorFlow and I\'m a little surprised on how much memory that is taking. On my GPU I can train YOLO using their Darknet framework with b

Answer 1

See the description in this (commit). The memory allocation is raw info is there although it needs a script to collect the information in an easy to read form.