ObjectDetector.jl icon indicating copy to clipboard operation
ObjectDetector.jl copied to clipboard

GPU allocations during each forward pass (causing memory strain)

Open IanButterworth opened this issue 5 years ago • 0 comments

If you run the following on a CUDA system, you'll see gradual increase in gpu memory allocation through the repeated runs. This causes big slowdowns on memory constrained systems when the swap file starts to be used.

using ObjectDetector
ObjectDetector.benchmark(select=[3]) #run yolov3-tiny

IanButterworth avatar Apr 30 '20 17:04 IanButterworth