From 160eddddc4e265d5ee59a38797c30720bf46cd7c Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 27 May 2018 13:53:42 +0000
Subject: [PATCH] Minor fix
---
src/avgpool_layer.c | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/src/avgpool_layer.c b/src/avgpool_layer.c
index 0feae71..b6932fe 100644
--- a/src/avgpool_layer.c
+++ b/src/avgpool_layer.c
@@ -4,7 +4,7 @@
avgpool_layer make_avgpool_layer(int batch, int w, int h, int c)
{
- fprintf(stderr, "Avgpool Layer: %d x %d x %d image\n", w,h,c);
+ fprintf(stderr, "avg %4d x%4d x%4d -> %4d\n", w, h, c, c);
avgpool_layer l = {0};
l.type = AVGPOOL;
l.batch = batch;
@@ -19,7 +19,11 @@
int output_size = l.outputs * batch;
l.output = calloc(output_size, sizeof(float));
l.delta = calloc(output_size, sizeof(float));
+ l.forward = forward_avgpool_layer;
+ l.backward = backward_avgpool_layer;
#ifdef GPU
+ l.forward_gpu = forward_avgpool_layer_gpu;
+ l.backward_gpu = backward_avgpool_layer_gpu;
l.output_gpu = cuda_make_array(l.output, output_size);
l.delta_gpu = cuda_make_array(l.delta, output_size);
#endif
--
Gitblit v1.10.0