video-classification icon indicating copy to clipboard operation
video-classification copied to clipboard

gpu memory issue 2

Open babbu3682 opened this issue 5 years ago • 0 comments

I already posted my problem before. but I still have a problem with the cuda memory issue.

I tried to run your model called "CRNN" that is training CNN and RNN from scratch.

I didn't any torch your code and just running. In the first epoch there is no error.

But after a few epochs, the Cuda memory issue also shows up.

How can you run the CRNN from scratch no using torch.no_grad?

Every epoch increase, the memory increase until CUDA out of error .

babbu3682 avatar May 14 '20 06:05 babbu3682