search

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 #25

Open babbu3682 opened 4 years ago

babbu3682 commented 4 years ago

I have a problem with the Encoder model. I wanted to make the CRNN from scratch so I build my custom CNN encoder model like "Efficientnet-b0", it is worked but after 2 steps the gpu memory errors show up. I think GPU memory is stacked every step because of "for() function" in forward(). So, how can I solve that problem?

1 epoch image

2 epoch image

3 epoch image

4 epoch image

as you can see above, the memory increase until CUDA out of error .....

HHTseng commented 4 years ago

Sorry that I was not able to conceive where the problem may be without knowing more details how your new model "Efficientnet-b0" is constructed. I do not recognize this type of error from my experience.