From 7520949d84bb410ebc22dece85181d725e03727a Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Wed, 8 Jun 2016 11:07:31 -0700 Subject: [PATCH] idk just in case --- Makefile | 4 ++-- ai2.mk | 2 +- src/convolutional_layer.c | 4 ++++ src/detection_layer.c | 2 +- 4 files changed, 8 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index 28a0d17a..366faef1 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ -GPU=0 +GPU=1 CUDNN=0 -OPENCV=0 +OPENCV=1 DEBUG=0 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 diff --git a/ai2.mk b/ai2.mk index b72e97b1..57edc893 100644 --- a/ai2.mk +++ b/ai2.mk @@ -1,7 +1,7 @@ GPU=0 CUDNN=0 OPENCV=0 -DEBUG=0 +DEBUG=1 AI2=1 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c index f0c312c9..af867e5e 100644 --- a/src/convolutional_layer.c +++ b/src/convolutional_layer.c @@ -8,6 +8,10 @@ #include #include +#ifdef AI2 +#include "xnor_layer.h" +#endif + #ifndef AI2 #define AI2 0 #endif diff --git a/src/detection_layer.c b/src/detection_layer.c index 1adda06b..e103b4ea 100644 --- a/src/detection_layer.c +++ b/src/detection_layer.c @@ -176,7 +176,7 @@ void forward_detection_layer(const detection_layer l, network_state state) } } - if(1){ + if(0){ float *costs = calloc(l.batch*locations*l.n, sizeof(float)); for (b = 0; b < l.batch; ++b) { int index = b*l.inputs;