From 2f62fe33c913cd9484fe7f2486889d12292c66e0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 07 Feb 2015 02:53:53 +0000
Subject: [PATCH] saving weight files as binaries, hell yeah
---
src/network.c | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/src/network.c b/src/network.c
index eb39054..2ec0881 100644
--- a/src/network.c
+++ b/src/network.c
@@ -42,8 +42,6 @@
return "none";
}
-
-
network make_network(int n, int batch)
{
network net;
@@ -61,7 +59,6 @@
return net;
}
-
void forward_network(network net, float *input, float *truth, int train)
{
int i;
@@ -78,7 +75,7 @@
}
else if(net.types[i] == CROP){
crop_layer layer = *(crop_layer *)net.layers[i];
- forward_crop_layer(layer, input);
+ forward_crop_layer(layer, train, input);
input = layer.output;
}
else if(net.types[i] == COST){
@@ -136,7 +133,6 @@
}
else if(net.types[i] == CONNECTED){
connected_layer layer = *(connected_layer *)net.layers[i];
- //secret_update_connected_layer((connected_layer *)net.layers[i]);
update_connected_layer(layer);
}
}
@@ -539,6 +535,9 @@
normalization_layer layer = *(normalization_layer *)net.layers[i];
return get_normalization_image(layer);
}
+ else if(net.types[i] == DROPOUT){
+ return get_network_image_layer(net, i-1);
+ }
else if(net.types[i] == CROP){
crop_layer layer = *(crop_layer *)net.layers[i];
return get_crop_image(layer);
--
Gitblit v1.10.0