From 08b757a0bf76efe8c76b453063a1bb19315bcaa6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 14 Jan 2015 20:18:57 +0000
Subject: [PATCH] Stable, needs to be way faster
---
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..341b80f 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, __global float *output)
{
int id = get_global_id(0);
- input[id] = (rand[id] < prob) ? 0 : input[id]/(1.-prob);
+ output[id] = (rand[id] < prob) ? 0 : input[id]*scale;
}
--
Gitblit v1.10.0