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

Function call stack: keras_scratch_graph #2

Open
saranggoel opened this issue Nov 23, 2020 · 0 comments
Open

Function call stack: keras_scratch_graph #2

saranggoel opened this issue Nov 23, 2020 · 0 comments

Comments

@saranggoel
Copy link

I really like your code and I am trying to run it with the given dataset. I am trying to use my NVIDIA gpu, but whenever I do, it runs into memory problems and raises the following error:

2020-11-22 21:40:39.870589: W tensorflow/core/common_runtime/bfc_allocator.cc:245] Allocator (GPU_0_bfc) ran out of memory trying to allocate 1.09GiB with freed_by_count=0. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.

Function call stack: keras_scratch_graph at the end.

However, when I reduce the batchsize, the error stops, but I would want to run this code with a larger batchsize. Do you know what may be the problem?

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

1 participant