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.h | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/dropout_layer.h b/src/dropout_layer.h
index 0a6f034..12a819e 100644
--- a/src/dropout_layer.h
+++ b/src/dropout_layer.h
@@ -8,8 +8,10 @@
float probability;
float scale;
float *rand;
+ float *output;
#ifdef GPU
cl_mem rand_cl;
+ cl_mem output_cl;
#endif
} dropout_layer;
--
Gitblit v1.10.0