From 0f1a31648c5292fa49b35eac90a2ee676d6c13e6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 31 Jan 2015 06:05:23 +0000
Subject: [PATCH] idk, probably something changed
---
src/network_kernels.cu | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/src/network_kernels.cu b/src/network_kernels.cu
index a009174..de8f659 100644
--- a/src/network_kernels.cu
+++ b/src/network_kernels.cu
@@ -58,7 +58,7 @@
}
else if(net.types[i] == CROP){
crop_layer layer = *(crop_layer *)net.layers[i];
- forward_crop_layer_gpu(layer, input);
+ forward_crop_layer_gpu(layer, train, input);
input = layer.output_gpu;
}
//printf("Forward %d %s %f\n", i, get_layer_string(net.types[i]), sec(clock() - time));
@@ -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