sys.getrefcount() of a LightningModule

[MGheini]

Hi,

When I run this piece of code

import sys
import torch
import pytorch_lightning as pl

a = torch.nn.Module()
b = pl.LightningModule()
print(sys.getrefcount(b))
print(sys.getrefcount(a))

I get

3
2

So basically LightningModule has one extra reference created compared to the conventional torch Moule. In my particular case, for instance, this results in memory issues: even if I explicitly do del b the memory is not released, and I believe this is the root of the issue. Can you advise me on why this is the case, and how I can consequently del a LightningModule?

Thanks!