HHTseng / video-classification

Tutorial for video classification/ action recognition using 3D CNN/ CNN+RNN on UCF101
940 stars 216 forks source link

gpu memory issue 2 #26

Open babbu3682 opened 4 years ago

babbu3682 commented 4 years ago

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 .