From cd3617b11d57789db59ed4dd042d9adb89b9404a Mon Sep 17 00:00:00 2001
From: Yantao Xie <leeward.xie@gmail.com>
Date: Sun, 1 Apr 2018 16:21:15 +0800
Subject: [PATCH] Set ArgMax's one_blob_only as true. (#325)

---
 src/layer/argmax.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/layer/argmax.cpp b/src/layer/argmax.cpp
index d6933387e..3ede2a0db 100644
--- a/src/layer/argmax.cpp
+++ b/src/layer/argmax.cpp
@@ -22,6 +22,7 @@ DEFINE_LAYER_CREATOR(ArgMax)
 
 ArgMax::ArgMax()
 {
+    one_blob_only = true;
 }
 
 int ArgMax::load_param(const ParamDict& pd)