OOM during inference

[hoonkai]

Hi

Anyone know how much GPU memory is needed to run the checkpoint panet_mask_step179999.pth? I've got 8GB but python tools/infer_simple.py --dataset coco2017 --cfg configs/panet/e2e_panet_R-50-FPN_2x_mask.yaml --load_ckpt panet_mask_step179999.pth gives OOM:

  File "/Volumes/Data/Projects/PANet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__
    result = self.forward(*input, **kwargs)
  File "/Volumes/Data/Projects/PANet/src/PANet/lib/modeling/FPN.py", line 439, in forward
    fpn_rpn_conv = F.relu(self.FPN_RPN_conv(bl_in), inplace=True)
  File "/Volumes/Data/Projects/PANet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__
    result = self.forward(*input, **kwargs)
  File "/Volumes/Data/Projects/PANet/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward
    self.padding, self.dilation, self.groups)
RuntimeError: cuda runtime error (2) : out of memory at /Volumes/Data/Projects/pytorch/src/pytorch/aten/src/THC/generic/THCStorage.cu:58