From dea022f1982074375a17dd54401d0acbfffbfa56 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Wed, 5 Sep 2018 16:36:33 +0900 Subject: [PATCH] [enco.caffe] Set AvgPool2D stride (#1346) With this commit, enco caffe frontend now sets the stride of AvgPool2D using a value specified in a given caffe model (instead of hard-coded one). Signed-off-by: Jonghyun Park --- contrib/enco/frontend/caffe/src/Frontend.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/contrib/enco/frontend/caffe/src/Frontend.cpp b/contrib/enco/frontend/caffe/src/Frontend.cpp index ab9ab95..7fc619a 100644 --- a/contrib/enco/frontend/caffe/src/Frontend.cpp +++ b/contrib/enco/frontend/caffe/src/Frontend.cpp @@ -319,8 +319,8 @@ enco::Bundle Frontend::load(void) const op->window()->vertical(spec.window_height()); op->window()->horizontal(spec.window_height()); - assert(spec.vertical_stride() == 1); - assert(spec.horizontal_stride() == 1); + op->stride()->vertical(spec.vertical_stride()); + op->stride()->horizontal(spec.horizontal_stride()); return op; }; -- 2.7.4