From d6fbe86e7a8c1bc389902c90c57ee7e80f5475b9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 16 Dec 2014 19:40:05 +0000
Subject: [PATCH] updates?
---
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