From f88baf4a3a756140cef3ca07be98cabb803d80ae Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 18 Dec 2014 23:46:45 +0000
Subject: [PATCH] 99 problems
---
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 729dbc4..341b80f 100644
--- a/src/dropout_layer.cl
+++ b/src/dropout_layer.cl
@@ -1,5 +1,5 @@
-__kernel void yoloswag420blazeit360noscope(__global float *input, __global float *rand, float prob, float scale)
+__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]*scale;
+ output[id] = (rand[id] < prob) ? 0 : input[id]*scale;
}
--
Gitblit v1.10.0