Inpainting: CUDA OOM and Inconsistent Tensor Sizes

[turbo]

For all example images+masks in the inpainting example, I get this error for CUDA:

RuntimeError: cuda runtime error (2) : out of memory at ~/pytorch/aten/src/THC/generic/THCStorage.cu:58

and for dtype = torch.FloatTensor I get this:

RuntimeError: inconsistent tensor sizes at ~/pytorch/aten/src/TH/generic/THTensorMath.c:2864

Python 2, macOS 10.12.6, CUDA 9.0.176, cudnn 9.0 v7

[turbo]

This does not occur on linux (no CUDA).

[haimiaozh]

I also got an erroron Linux(no CUDA) when run the inpainting: RuntimeError: inconsistent tensor size at /opt/conda/conda-bld/libtorch_1493851953343/work/pytorch-0.1.12/torch/lib/TH/generic/THTensorCopy.c:51