From af4e4f92dc9e5da160eb6c6870a7b38b863f1c6c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 28 Oct 2014 02:45:06 +0000
Subject: [PATCH] getting rid of sub_arrays, nvidia driver memory leak

---
 src/network.h |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/src/network.h b/src/network.h
index c95f6fa..7625904 100644
--- a/src/network.h
+++ b/src/network.h
@@ -42,6 +42,7 @@
 cl_mem get_network_output_cl_layer(network net, int i);
 cl_mem get_network_delta_cl_layer(network net, int i);
 float train_network_sgd_gpu(network net, data d, int n);
+float train_network_data_gpu(network net, data d, int n);
 #endif
 
 network make_network(int n, int batch);

--
Gitblit v1.10.0