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

idk just in case

Browse Source
This commit is contained in:
Joseph Redmon 2016-06-08 11:07:31 -07:00
parent 8a767f1066
commit 7520949d84
4 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Makefile
View File

@ -1,6 +1,6 @@
GPU=0 GPU=1
CUDNN=0 CUDNN=0
OPENCV=0 OPENCV=1
DEBUG=0 DEBUG=0
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52

2
ai2.mk
View File

@ -1,7 +1,7 @@
GPU=0 GPU=0
CUDNN=0 CUDNN=0
OPENCV=0 OPENCV=0
DEBUG=0 DEBUG=1
AI2=1 AI2=1
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52

4
src/convolutional_layer.c
View File

@ -8,6 +8,10 @@
#include <stdio.h> #include <stdio.h>
#include <time.h> #include <time.h>
#ifdef AI2
#include "xnor_layer.h"
#endif
#ifndef AI2 #ifndef AI2
#define AI2 0 #define AI2 0
#endif #endif

2
src/detection_layer.c
View File

@ -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)); float *costs = calloc(l.batch*locations*l.n, sizeof(float));
for (b = 0; b < l.batch; ++b) { for (b = 0; b < l.batch; ++b) {
int index = b*l.inputs; int index = b*l.inputs;