tensorflow-yolo-v3
tensorflow-yolo-v3 copied to clipboard
Running out of memory error when running demo
Anyway to reduce the memory usage when running the demo? the error shows it requires 5.93 GB to run the demo with gpu version of tensorflow. My system only has 5.5 GB. Is this requirement normal? or I did something wrong there?
If anyone could help, please. Thanks
Quick fix is to reduce default value of gpu_memory_fraction in line 42 (demo.py). Set something like 0.8 and it should work.
Quick fix is to reduce default value of gpu_memory_fraction in line 42 (demo.py). Set something like 0.8 and it should work.
Thank you !!