[enco.caffe] Set AvgPool2D stride (#1346)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Wed, 5 Sep 2018 07:36:33 +0000 (16:36 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Wed, 5 Sep 2018 07:36:33 +0000 (16:36 +0900)
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 <jh1302.park@samsung.com>
contrib/enco/frontend/caffe/src/Frontend.cpp

index ab9ab95..7fc619a 100644 (file)
@@ -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;
       };