From db0397cfaaf488364e3d2e1669dfefae2ee6ea73 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Mon, 14 Dec 2015 19:57:10 +0000 Subject: [PATCH] shortcut layers, msr networks --- src/maxpool_layer.h | 3 ++- 1 files changed, 2 insertions(+), 1 deletions(-) diff --git a/src/maxpool_layer.h b/src/maxpool_layer.h index 4456863..b91c2c1 100644 --- a/src/maxpool_layer.h +++ b/src/maxpool_layer.h @@ -5,12 +5,13 @@ #include "params.h" #include "cuda.h" #include "layer.h" +#include "network.h" typedef layer maxpool_layer; image get_maxpool_image(maxpool_layer l); maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int stride); -void resize_maxpool_layer(maxpool_layer *l, int h, int w); +void resize_maxpool_layer(maxpool_layer *l, int w, int h); void forward_maxpool_layer(const maxpool_layer l, network_state state); void backward_maxpool_layer(const maxpool_layer l, network_state state); -- Gitblit v1.10.0