From 2b2441313b73c460a60c013c3b7bf9e19c994b6b Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 30 Oct 2014 18:28:37 +0000 Subject: [PATCH] col2im maybe a little faster --- 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