From 90d354a2a5a3ba76071337d8794cfc00f7bc5fab Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 13 Dec 2014 20:01:21 +0000
Subject: [PATCH] fixed dropout ><
---
src/dropout_layer.cl | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/dropout_layer.cl b/src/dropout_layer.cl
index aa24964..729dbc4 100644
--- a/src/dropout_layer.cl
+++ b/src/dropout_layer.cl
@@ -1,5 +1,5 @@
-__kernel void forward(__global float *input, __global float *rand, float prob)
+__kernel void yoloswag420blazeit360noscope(__global float *input, __global float *rand, float prob, float scale)
{
int id = get_global_id(0);
- input[id] = (rand[id] < prob) ? 0 : input[id]/(1.-prob);
+ input[id] = (rand[id] < prob) ? 0 : input[id]*scale;
}
--
Gitblit v1.10.0