CUDA OUT OF MEMORY

[Zhangheng-Fighter]

只有将max_im_shrink设为1的时候才不会溢出，1080Ti的环境，感觉是什么东西一下子占了很多显存，虽然只allocate不到2G，但free的显存很少。