Browse Source

convert LogisticRegressionOutput

tags/20180704
nihuini 8 years ago
parent
commit
307a77f04b
1 changed files with 9 additions and 2 deletions
  1. +9
    -2
      tools/mxnet/mxnet2ncnn.cpp

+ 9
- 2
tools/mxnet/mxnet2ncnn.cpp View File

@@ -939,6 +939,10 @@ int main(int argc, char** argv)
{
fprintf(pp, "%-16s", "UnaryOp");
}
else if (n.op == "LogisticRegressionOutput")
{
fprintf(pp, "%-16s", "Sigmoid");
}
else if (n.op == "max")
{
fprintf(pp, "%-16s", "Reduction");
@@ -1040,7 +1044,7 @@ int main(int argc, char** argv)
}
}

if (n.op == "SoftmaxOutput")
if (n.op == "SoftmaxOutput" || n.op == "LogisticRegressionOutput")
{
// drop label
input_size--;
@@ -1057,7 +1061,7 @@ int main(int argc, char** argv)
continue;
}

if (n.op == "SoftmaxOutput")
if (n.op == "SoftmaxOutput" || n.op == "LogisticRegressionOutput")
{
// drop label
if (j == 1)
@@ -1549,6 +1553,9 @@ int main(int argc, char** argv)
int op_type = 8;
fprintf(pp, " 0=%d", op_type);
}
else if (n.op == "LogisticRegressionOutput")
{
}
else if (n.op == "max")
{
int operation = 4;


Loading…
Cancel
Save