pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:53: lambda [](int)-＞auto::opera

GPU上运行神经网络模型，是使用多线程并行加速的，如果出错，真正的错误信息很容易被cuda的错误信息淹没，所以难以看到错误信息，一般引起问题的原因是数组等超限，定位到具体是anchor_ratios=[0.2, 0.5, 1.0, 2.0, 5.0]修改成默认anchor_ratios=[0.5, 1.0, 2.0]就可以啦

pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:53: lambda [](int)-＞auto::opera

浏览过的版块