From 153705226d8ca746478b69eeac9bc854766daa11 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 27 Jan 2015 21:31:06 +0000
Subject: [PATCH] Bias updates bug fix
---
src/network_kernels.cu | 9 +++++++++
1 files changed, 9 insertions(+), 0 deletions(-)
diff --git a/src/network_kernels.cu b/src/network_kernels.cu
index a009174..7909e46 100644
--- a/src/network_kernels.cu
+++ b/src/network_kernels.cu
@@ -176,6 +176,7 @@
float train_network_datum_gpu(network net, float *x, float *y)
{
+ //clock_t time = clock();
int x_size = get_network_input_size(net)*net.batch;
int y_size = get_network_output_size(net)*net.batch;
if(!*net.input_gpu){
@@ -185,10 +186,18 @@
cuda_push_array(*net.input_gpu, x, x_size);
cuda_push_array(*net.truth_gpu, y, y_size);
}
+ //printf("trans %f\n", sec(clock() - time));
+ //time = clock();
forward_network_gpu(net, *net.input_gpu, *net.truth_gpu, 1);
+ //printf("forw %f\n", sec(clock() - time));
+ //time = clock();
backward_network_gpu(net, *net.input_gpu);
+ //printf("back %f\n", sec(clock() - time));
+ //time = clock();
update_network_gpu(net);
float error = get_network_cost(net);
+ //printf("updt %f\n", sec(clock() - time));
+ //time = clock();
return error;
}
--
Gitblit v1.10.0