gaopengcuhk/Stable-Pix2Seq

CUDA out of memory during training

yshMars opened this issue · 0 comments

I was training 'Stable Pix2Seq', everything goes fine until the 3rd training epoch. I wonder if there's any accumulate operation or some tensors or variable should have been deleted.