fix fully_connected

89efdf4e · yejianwu · 49e1e25b · 89efdf4e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mace/ops/fully_connected.h mace/ops/fully_connected.h +1 -1

未找到文件。
--- a/mace/ops/fully_connected.h
+++ b/mace/ops/fully_connected.h
@@ -30,7 +30,7 @@ class FullyConnectedOp : public Operator<D, T> {
    : Operator<D, T>(operator_def, ws),
      functor_(OperatorBase::GetSingleArgument<int>(
                   "weight_type",
-                   7 /*static_cast<int>(kernels::WEIGHT_WIDTH)*/),
+                   8 /*static_cast<int>(kernels::WEIGHT_WIDTH)*/),
               kernels::StringToActivationType(
                 OperatorBase::GetSingleArgument<std::string>("activation",
                                                              "NOOP")),