CornerNet-Lite
CornerNet-Lite copied to clipboard
Device index must be -1 or non-negative, got -1160
Anyone met this problem?
Maybe there is some problem with your GPU setting.
I met the same problem. The default number of gpu is 4. But i only have one gpu, so i set cuda:0. I wonder whether the problem is about this setting.
if you only has one gpu the chunk_size should look this chunk_size =[15] batch_size = 15