From 8561e49b5a2876e9a522b2dedfa99f19d5738154 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 13 Jul 2015 22:04:21 +0000
Subject: [PATCH] add avgpool layer
---
src/normalization_layer.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/normalization_layer.c b/src/normalization_layer.c
index dce2fcc..d080559 100644
--- a/src/normalization_layer.c
+++ b/src/normalization_layer.c
@@ -40,10 +40,10 @@
layer->out_w = w;
layer->inputs = w*h*c;
layer->outputs = layer->inputs;
- layer->output = realloc(layer->output, h * w * layer->c * layer->batch * sizeof(float));
- layer->delta = realloc(layer->delta, h * w * layer->c * layer->batch * sizeof(float));
- layer->squared = realloc(layer->squared, h * w * layer->c * layer->batch * sizeof(float));
- layer->norms = realloc(layer->norms, h * w * layer->c * layer->batch * sizeof(float));
+ layer->output = realloc(layer->output, h * w * c * batch * sizeof(float));
+ layer->delta = realloc(layer->delta, h * w * c * batch * sizeof(float));
+ layer->squared = realloc(layer->squared, h * w * c * batch * sizeof(float));
+ layer->norms = realloc(layer->norms, h * w * c * batch * sizeof(float));
#ifdef GPU
cuda_free(layer->output_gpu);
cuda_free(layer->delta_gpu);
--
Gitblit v1.10.0