菜鸟 学习faster_rcnn 之跑通网络(补充)
出现RuntimeError: cuda runtime error (2) : out of memory at …\aten\src\THC\THCCachingHostAllocator.cpp:278这个错误
需要更改
参考这里
菜鸟 学习faster_rcnn 之跑通网络(补充)
出现RuntimeError: cuda runtime error (2) : out of memory at …\aten\src\THC\THCCachingHostAllocator.cpp:278这个错误
需要更改
参考这里