Skip to content

Gpu memory usage #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
brkyc3 opened this issue Mar 26, 2019 · 1 comment
Open

Gpu memory usage #3

brkyc3 opened this issue Mar 26, 2019 · 1 comment

Comments

@brkyc3
Copy link

brkyc3 commented Mar 26, 2019

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
Copy link
Owner

OValery16 commented May 4, 2019

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants