From d7286c273211ffeb1f56594f863d1ee9922be6d4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 07 Nov 2013 00:09:41 +0000
Subject: [PATCH] Loading may or may not work. But probably.
---
src/maxpool_layer.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/maxpool_layer.c b/src/maxpool_layer.c
index 38ac582..6f7d2a2 100644
--- a/src/maxpool_layer.c
+++ b/src/maxpool_layer.c
@@ -1,10 +1,10 @@
#include "maxpool_layer.h"
-maxpool_layer make_maxpool_layer(int h, int w, int c, int stride)
+maxpool_layer *make_maxpool_layer(int h, int w, int c, int stride)
{
- maxpool_layer layer;
- layer.stride = stride;
- layer.output = make_image((h-1)/stride+1, (w-1)/stride+1, c);
+ maxpool_layer *layer = calloc(1, sizeof(maxpool_layer));
+ layer->stride = stride;
+ layer->output = make_image((h-1)/stride+1, (w-1)/stride+1, c);
return layer;
}
--
Gitblit v1.10.0