Fixed compile error

pull/492/head
AlexeyAB 7 years ago
parent e947bf29af
commit 2f52cfeb07
  1. 2
      src/yolo_v2_class.cpp

@ -22,12 +22,14 @@ extern "C" {
#define FRAMES 3
#ifdef GPU
void check_cuda(cudaError_t status) {
if (status != cudaSuccess) {
const char *s = cudaGetErrorString(status);
printf("CUDA Error Prev: %s\n", s);
}
}
#endif
struct detector_gpu_t {
float **probs;

Loading…
Cancel
Save