From 7e9e289b164e9bbfb77f644b1cbfac48bc8c8408 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Tue, 11 Sep 2018 00:51:08 +0000
Subject: [PATCH] training w/ full yolo cfg
---
src/maxpool_layer.c | 6 ++++++
1 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/src/maxpool_layer.c b/src/maxpool_layer.c
index c543b52..5ff7e9d 100644
--- a/src/maxpool_layer.c
+++ b/src/maxpool_layer.c
@@ -1,5 +1,6 @@
#include "maxpool_layer.h"
#include "cuda.h"
+#include "gemm.h"
#include <stdio.h>
image get_maxpool_image(maxpool_layer l)
@@ -79,6 +80,11 @@
void forward_maxpool_layer(const maxpool_layer l, network_state state)
{
+ if (!state.train) {
+ forward_maxpool_layer_avx(state.input, l.output, l.indexes, l.size, l.w, l.h, l.out_w, l.out_h, l.c, l.pad, l.stride, l.batch);
+ return;
+ }
+
int b,i,j,k,m,n;
int w_offset = -l.pad / 2;
int h_offset = -l.pad / 2;
--
Gitblit v1.10.0