max_im_shrink

[ljmkgbkq]

请问在哪里设置max_im_shrink

[Llq201809]

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.