search

OValery16 / Tutorial-about-3D-convolutional-network

Tutorial about 3D convolutional network
Mozilla Public License 2.0
227 stars 36 forks source link

Gpu memory usage #3

Open brkyc3 opened 5 years ago

brkyc3 commented 5 years ago

Hi thanks for tutorial , i want to try this, with part of 20bn-jester data on colab.But even though i use batch size of 64 , i getting OOM error on GPU. What hardware do you use for training and how long it takes to complete 1 epoch on all dataset ?

OValery16 commented 5 years ago

I used 4 1080TI GPU. I would like to remind you something important. In contrast to RGB images, videos require significantly more space. 1 batch = size of the batch number of frames in your video size of the image (W H C). For instance; a batch size of 64 videos requires significantly more memory than a batch of 64 images.