PoinTr icon indicating copy to clipboard operation
PoinTr copied to clipboard

memory leak in the training code

Open cckamiya opened this issue 1 year ago • 1 comments

There is a memory leak in the training code of AdaPoinTr_PCN model when I train my own model, and my training memory of more than 2200epoch can occupy about 40G+.By the memory, I don't mean the graphics memory of the graphics card, but the memory of the computer.

cckamiya avatar Aug 28 '23 09:08 cckamiya

Hey @cckamiya, it appears that we're saving the picture during each validation here. If you believe the visualization in TFBorad isn't needed, go ahead and comment out this section of the code to see if it resolves the issue.

yuxumin avatar Aug 28 '23 09:08 yuxumin