Error appers when I use "Detect Objects Using Deep Learning" in ArcGIS Pro, the model is usa_building_footprints.dlpk which trained by ArcGIS Pro 2.7.3.
Expected behavior
How to make the model work.
Platform (please complete the following information):
Solved! Go to Solution.
It is the error directly above that is causing the problem: RuntimeError: CUDA out of memory.
This means that all the dedicated memory on your graphics card is being used up while running the Detect Objects tool. The Batch Size parameter controls how many image segments are sent to the graphics card at once for detection. You will need to reduce the Batch Size parameter in the tool until it runs without error. I suggest setting it to 1 or 2 to begin with, and you can increase it later if required. See attached screenshot to see where the Batch Size parameter can be found.
It is the error directly above that is causing the problem: RuntimeError: CUDA out of memory.
This means that all the dedicated memory on your graphics card is being used up while running the Detect Objects tool. The Batch Size parameter controls how many image segments are sent to the graphics card at once for detection. You will need to reduce the Batch Size parameter in the tool until it runs without error. I suggest setting it to 1 or 2 to begin with, and you can increase it later if required. See attached screenshot to see where the Batch Size parameter can be found.