mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed compile error

Browse Source
This commit is contained in:
AlexeyAB
2018-03-17 01:12:00 +03:00
parent e947bf29af
commit 2f52cfeb07
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/yolo_v2_class.cpp
View File

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