From d6fbe86e7a8c1bc389902c90c57ee7e80f5475b9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 16 Dec 2014 19:40:05 +0000
Subject: [PATCH] updates?
---
src/network_gpu.c | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/src/network_gpu.c b/src/network_gpu.c
index d09aa71..c3f22d3 100644
--- a/src/network_gpu.c
+++ b/src/network_gpu.c
@@ -55,6 +55,11 @@
dropout_layer layer = *(dropout_layer *)net.layers[i];
forward_dropout_layer_gpu(layer, input);
}
+ else if(net.types[i] == CROP){
+ crop_layer layer = *(crop_layer *)net.layers[i];
+ forward_crop_layer_gpu(layer, input);
+ input = layer.output_cl;
+ }
//printf("%d %f\n", i, sec(clock()-time));
/*
else if(net.types[i] == CROP){
@@ -142,6 +147,10 @@
maxpool_layer layer = *(maxpool_layer *)net.layers[i];
return layer.output_cl;
}
+ else if(net.types[i] == CROP){
+ crop_layer layer = *(crop_layer *)net.layers[i];
+ return layer.output_cl;
+ }
else if(net.types[i] == SOFTMAX){
softmax_layer layer = *(softmax_layer *)net.layers[i];
return layer.output_cl;
@@ -260,7 +269,7 @@
float *network_predict_gpu(network net, float *input)
{
-
+
int size = get_network_input_size(net) * net.batch;
cl_mem input_cl = cl_make_array(input, size);
forward_network_gpu(net, input_cl, 0, 0);
--
Gitblit v1.10.0