From 19d3ae7267c355414a6207835336a3b40d5b053a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 18 Dec 2014 21:21:30 +0000
Subject: [PATCH] message
---
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