nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

Set ArgMax's one_blob_only as true. (#325)

tags/20180427
Yantao Xie nihui 8 years ago
parent
9748c00b44
commit
cd3617b11d
1 changed files with 1 additions and 0 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      src/layer/argmax.cpp

+ 1
- 0
src/layer/argmax.cpp View File

@@ -22,6 +22,7 @@ DEFINE_LAYER_CREATOR(ArgMax)

ArgMax::ArgMax()
{
one_blob_only = true;
}

int ArgMax::load_param(const ParamDict& pd)


Write Preview
Loading…
Cancel
Save