yxlijun/Pyramidbox.pytorch

max_im_shrink

ljmkgbkq opened this issue · 1 comments

ljmkgbkq commented

请问在哪里设置max_im_shrink

Llq201809 commented

you don't need to modify max_im_shrink if out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58, just change
x = Variable(torch.from_numpy(x).unsqueeze(0)) to :
x = Variable(torch.from_numpy(x).unsqueeze(0),volatile=True), it works.