COVID-CT-Code icon indicating copy to clipboard operation
COVID-CT-Code copied to clipboard

Function call stack: keras_scratch_graph

Open saranggoel opened this issue 3 years ago • 0 comments

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?

saranggoel avatar Nov 23 '20 04:11 saranggoel