From d97331b88ff3d50035b1e22c9d0eb671b61227e3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 15 Apr 2015 07:32:32 +0000
Subject: [PATCH] level adjustment for images
---
src/network_kernels.cu | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/src/network_kernels.cu b/src/network_kernels.cu
index 019f40d..0b2bb97 100644
--- a/src/network_kernels.cu
+++ b/src/network_kernels.cu
@@ -71,6 +71,7 @@
state.input = get_network_output_gpu_layer(net, i-1);
state.delta = get_network_delta_gpu_layer(net, i-1);
}
+
if(net.types[i] == CONVOLUTIONAL){
backward_convolutional_layer_gpu(*(convolutional_layer *)net.layers[i], state);
}
--
Gitblit v1.10.0