COVID-CT-Code
COVID-CT-Code copied to clipboard
Function call stack: keras_scratch_graph
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?