笔记Hint: If you want to see a list of allocated tensors when OOM

Faster-rcnn 爆存问题

前言

在根据睿智博主大佬的检测算法复现自己的数据集时，发现哎呀，内存不够啦
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

一、解决方法？

我发现把self.num_rois改小就可以跑起来啦，不知道这种解决方法有理论依据不，有大佬知道吗？