From 5f4a5f59b072d4029107422d30b04941424c48b1 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 24 Feb 2015 02:52:05 +0000
Subject: [PATCH] captcha stuff
---
src/cost_layer.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/cost_layer.c b/src/cost_layer.c
index a08562b..34c8fb5 100644
--- a/src/cost_layer.c
+++ b/src/cost_layer.c
@@ -49,7 +49,7 @@
if(layer.type == DETECTION){
int i;
for(i = 0; i < layer.batch*layer.inputs; ++i){
- if((i%5) && !truth[(i/5)*5]) layer.delta[i] = 0;
+ if((i%25) && !truth[(i/25)*25]) layer.delta[i] = 0;
}
}
*(layer.output) = dot_cpu(layer.batch*layer.inputs, layer.delta, 1, layer.delta, 1);
@@ -71,7 +71,7 @@
axpy_ongpu(layer.batch*layer.inputs, -1, input, 1, layer.delta_gpu, 1);
if(layer.type==DETECTION){
- mask_ongpu(layer.inputs*layer.batch, layer.delta_gpu, truth, 5);
+ mask_ongpu(layer.inputs*layer.batch, layer.delta_gpu, truth, 25);
}
cuda_pull_array(layer.delta_gpu, layer.delta, layer.batch*layer.inputs);
--
Gitblit v1.10.0