Skip to content
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

Question about speed and memory? #5

Open
SunGaofeng opened this issue Oct 21, 2019 · 1 comment
Open

Question about speed and memory? #5

SunGaofeng opened this issue Oct 21, 2019 · 1 comment

Comments

@SunGaofeng
Copy link

Thanks to your great week on this repo, I have some questions about speed and memory of this model,
1, Do you use a single V100 to train this model? How long does it take to complete on the two dataset, moving mnist and KTH actions?
2, In your default settings, batch_size = 4, why use such a small batch_size? Is it because that the model is too large? What's the memory of your V100, 32G or 16G?

@GuptaSonam
Copy link

Hi SunGaofeng,

Did you try running the code? For me, training for MNIST dataset on Ti gtx 1080 (12 gb) takes 1 week using a single gpu. Not sure if I am doing something wrong or the model is supposed to be that slow.

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