Stable-Pix2Seq icon indicating copy to clipboard operation
Stable-Pix2Seq copied to clipboard

Published 20 hours ago • verified publisher icon gaopengcuhk

CUDA out of memory during training

Open yshMars opened this issue 2 years ago • 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.

yshMars avatar Mar 08 '22 02:03 yshMars