TensorRT-CenterNet icon indicating copy to clipboard operation
TensorRT-CenterNet copied to clipboard

Issue when applying mobileNet

Open Jumponthemoon opened this issue 5 years ago • 1 comments

Hi,i tried the mobilenetV2 you provided in the other repo.It successfully converted to the desired engine.But when i run the runDet command and evaluation command.It throws out the exception:

terminate called after throwing an instance of 'thrust::system::system_error' what(): parallel_for failed: no kernel image is available for execution on the device

I don't know where the problem came from.Anybody could help?

Jumponthemoon avatar Jul 06 '20 06:07 Jumponthemoon

Btw the mobilenet setting as get_mobile_net(5,{'hm':18,'reg':2,'wh':2}),the head_conv set as default 64. I'm not sure if the head_conv number is too large or should i set the head_conv as 24

Jumponthemoon avatar Jul 06 '20 06:07 Jumponthemoon