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

Is there any memory-efficient tensorflow Implementation? #32

Open
feynman233 opened this issue Aug 27, 2017 · 2 comments
Open

Is there any memory-efficient tensorflow Implementation? #32

feynman233 opened this issue Aug 27, 2017 · 2 comments

Comments

@feynman233
Copy link

Hi, I have found the same problem about the GPU memory,and read your report(https://arxiv.org/pdf/1707.06990.pdf).
So, is there any memory-efficient tensorflow Implementation?
Thanks very much!

@liuzhuang13
Copy link
Owner

Sorry we don't have any memory-efficient tensorflow implementation, as far as I know.

But I think for practical usage, you don't always need a memory-efficient implementation. Note that memory-efficient version still costs you at least the same amout of time. In most of time, if you have a 4-GPU machine, naive implementation is enough. Thanks!

@joeyearsley
Copy link

https://github.com/joeyearsley/efficient_densenet_tensorflow

Made using gradient checkpointing like the gpleiss repo does.

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

3 participants