From afe245329bdd14f36ca773fd7d497fc628b0535a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 10 Jun 2015 18:14:04 +0000
Subject: [PATCH] Better load messaging
---
src/maxpool_layer.c | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/maxpool_layer.c b/src/maxpool_layer.c
index c7739f1..159bd0a 100644
--- a/src/maxpool_layer.c
+++ b/src/maxpool_layer.c
@@ -27,14 +27,14 @@
l.h = h;
l.w = w;
l.c = c;
- l.out_h = h;
- l.out_w = w;
+ l.out_h = (h-1)/stride + 1;
+ l.out_w = (w-1)/stride + 1;
l.out_c = c;
l.outputs = l.out_h * l.out_w * l.out_c;
l.inputs = l.outputs;
l.size = size;
l.stride = stride;
- int output_size = ((h-1)/stride+1) * ((w-1)/stride+1) * c * batch;
+ int output_size = l.out_h * l.out_w * l.out_c * batch;
l.indexes = calloc(output_size, sizeof(int));
l.output = calloc(output_size, sizeof(float));
l.delta = calloc(output_size, sizeof(float));
--
Gitblit v1.10.0