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.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/network.c b/src/network.c
index eb39054..b628561 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){
@@ -539,6 +536,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