From 7756cccb793bb4950c241f2804195ea859d1b407 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 14 Oct 2014 05:31:48 +0000
Subject: [PATCH] Refactored connected to use blas
---
src/network.c | 12 ++++++++++++
1 files changed, 12 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 5833166..e4e4c8e 100644
--- a/src/network.c
+++ b/src/network.c
@@ -229,6 +229,8 @@
return layer.output;
} else if(net.types[i] == DROPOUT){
return get_network_output_layer(net, i-1);
+ } else if(net.types[i] == FREEWEIGHT){
+ return get_network_output_layer(net, i-1);
} else if(net.types[i] == CONNECTED){
connected_layer layer = *(connected_layer *)net.layers[i];
return layer.output;
@@ -258,6 +260,8 @@
return layer.delta;
} else if(net.types[i] == DROPOUT){
return get_network_delta_layer(net, i-1);
+ } else if(net.types[i] == FREEWEIGHT){
+ return get_network_delta_layer(net, i-1);
} else if(net.types[i] == CONNECTED){
connected_layer layer = *(connected_layer *)net.layers[i];
return layer.delta;
@@ -424,6 +428,10 @@
dropout_layer layer = *(dropout_layer *) net.layers[i];
return layer.inputs;
}
+ else if(net.types[i] == FREEWEIGHT){
+ freeweight_layer layer = *(freeweight_layer *) net.layers[i];
+ return layer.inputs;
+ }
else if(net.types[i] == SOFTMAX){
softmax_layer layer = *(softmax_layer *)net.layers[i];
return layer.inputs;
@@ -451,6 +459,10 @@
dropout_layer layer = *(dropout_layer *) net.layers[i];
return layer.inputs;
}
+ else if(net.types[i] == FREEWEIGHT){
+ freeweight_layer layer = *(freeweight_layer *) net.layers[i];
+ return layer.inputs;
+ }
else if(net.types[i] == SOFTMAX){
softmax_layer layer = *(softmax_layer *)net.layers[i];
return layer.inputs;
--
Gitblit v1.10.0