I recently updated the pytorch to 0.4.0, and found some wierd thing.
when I forward the Tensor to the MaxUnPool3D, the GPU memory keep increasing.
like,
x = torch.rand(1,1,144,144,144).to('cuda:0')
mp = torch.nn.MaxPool3d(2,2, return_indices=True)
mup = torch.nn.MaxUnpool3d(2,2)
x, i = mp(x)
# 579 Mib
a = mup(x,i)
# 579 Mib
a = mup(x,i)
# 591 Mib
a = mup(x,i)
# 601 Mib
...

Is there I miss something or is the bug?