Giters

Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Where should I change the number of GPUs used?

ChenJiaDeYang opened this issue · comments

commented

ERROR:--->>>CUDA out of memory. Tried to allocate 1.02 GiB (GPU 0; 15.75 GiB total capacity; 13.79 GiB already allocated; 261.62 MiB free; 570.45 MiB cached)