Interesting that “del” helps here - could eg torch.jit.script/trace be more of a “sledgehammer” solution?