Hi, when I compile your project, error occurs due to cudnn version problem, so I compile without cudnn. So first I am wondering what's your cudnn version?
My model in tiny-yolov2, got 1 class to detect, when I train my model, error ocurrs:
''F0928 12:13:05.047139 1687 region_loss_layer.cpp:182] Check failed: input_count == tmp_input_count (4320 vs. 5070) ''
So what does '30' mean in region_loss_layer.cpp, line 181? What else code should I change? Thanks a lot
Hi, when I compile your project, error occurs due to cudnn version problem, so I compile without cudnn. So first I am wondering what's your cudnn version? My model in tiny-yolov2, got 1 class to detect, when I train my model, error ocurrs: ''F0928 12:13:05.047139 1687 region_loss_layer.cpp:182] Check failed: input_count == tmp_input_count (4320 vs. 5070) '' So what does '30' mean in region_loss_layer.cpp, line 181? What else code should I change? Thanks a lot