Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 89efdf4e 编写于 作者: Y yejianwu
浏览文件
操作

fix fully_connected

上级 49e1e25b
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
mace/ops/fully_connected.h
浏览文件 @ 89efdf4e
...@@ -30,7 +30,7 @@ class FullyConnectedOp : public Operator<D, T> { ...@@ -30,7 +30,7 @@ class FullyConnectedOp : public Operator<D, T> {
: Operator<D, T>(operator_def, ws), : Operator<D, T>(operator_def, ws),
functor_(OperatorBase::GetSingleArgument<int>( functor_(OperatorBase::GetSingleArgument<int>(
"weight_type", "weight_type",
7 /*static_cast<int>(kernels::WEIGHT_WIDTH)*/), 8 /*static_cast<int>(kernels::WEIGHT_WIDTH)*/),
kernels::StringToActivationType( kernels::StringToActivationType(
OperatorBase::GetSingleArgument<std::string>("activation", OperatorBase::GetSingleArgument<std::string>("activation",
"NOOP")), "NOOP")),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或