From 89efdf4ecd736f5d236c6ac02fd7391776709dd2 Mon Sep 17 00:00:00 2001 From: yejianwu Date: Wed, 25 Apr 2018 17:19:42 +0800 Subject: [PATCH] fix fully_connected --- mace/ops/fully_connected.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/ops/fully_connected.h b/mace/ops/fully_connected.h index 26713d9a..255acee4 100644 --- a/mace/ops/fully_connected.h +++ b/mace/ops/fully_connected.h @@ -30,7 +30,7 @@ class FullyConnectedOp : public Operator { : Operator(operator_def, ws), functor_(OperatorBase::GetSingleArgument( "weight_type", - 7 /*static_cast(kernels::WEIGHT_WIDTH)*/), + 8 /*static_cast(kernels::WEIGHT_WIDTH)*/), kernels::StringToActivationType( OperatorBase::GetSingleArgument("activation", "NOOP")), -- GitLab