Hello!
I am trying to train (to detect my custom objects):
I have performed 1-7 steps according to what you said. When I am ready to enter the command line excitedly, here comes "CUDA ERROR: out of memory",
My environment is MSVS 2015, CUDA 9.1 and OpenCV 3.0.0. My coding ability is poor. I'm sorry for asking a question that may be relatively simple, Thank you so much!
The command line output is as follows:
D:Workplacedarknetdarknet-Wangbuilddarknetx64>darknet.exe detector train D:Workplacedarknetdarknet-Wangbuilddarknetx64dataobj.data D:Workplacedarknetdarknet-Wangbuilddarknetx64yolo-obj.cfg D:Workplacedarknetdarknet-Wangbuilddarknetx64darknet19_448.conv.23
yolo-obj
layer filters size input output
0 conv 32 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 32
1 max 2 x 2 / 2 416 x 416 x 32 -> 208 x 208 x 32
2 conv 64 3 x 3 / 1 208 x 208 x 32 -> 208 x 208 x 64
3 max 2 x 2 / 2 208 x 208 x 64 -> 104 x 104 x 64
4 conv 128 3 x 3 / 1 104 x 104 x 64 -> 104 x 104 x 128
5 conv 64 1 x 1 / 1 104 x 104 x 128 -> 104 x 104 x 64
6 CUDA Error: out of memory
CUDA Error: out of memory: No error
read readme carefully.
if error Out of memory occurs then in .cfg-file you should increase subdivisions=16, 32 or 64
Sorry, I didn't read it carefully. Thank you very much!
you should close all issues about "CUDA Error: out of memory"
Most helpful comment
read readme carefully.