Why don’t you use Tensorboard profiler. You can use the op stack and memory usage to understand how much memory each op uses
Why don’t you use Tensorboard profiler. You can use the op stack and memory usage to understand how much memory each op uses