I have this issue with trt_->doinfer(buffers);
BarryLoveBerry opened this issue · comments
Hi all,
I just wondering those parameters in below is the GPU memory or CPU's?
void *buffers[] = {features_input_, voxel_idxs_, params_input_, cls_output_, box_output_, dir_cls_output_};
trt_->doinfer(buffers);