From ca3708c3e157ea7fc21cd02ec7a91f80e5c42c27 Mon Sep 17 00:00:00 2001
From: SunAhong1993 <sunyanfang01@yq01-gpu-255-134-17-00.epc.baidu.com>
Date: Wed, 11 Sep 2019 10:39:06 +0800
Subject: [PATCH] fix the bug

---
 caffe2fluid/kaffe/net_template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/caffe2fluid/kaffe/net_template.py b/caffe2fluid/kaffe/net_template.py
index f9387c9..c6810c6 100644
--- a/caffe2fluid/kaffe/net_template.py
+++ b/caffe2fluid/kaffe/net_template.py
@@ -100,7 +100,7 @@ def main():
     npy_weight = args.npy_path
     fluid_model = args.model_param_path
     outputs = None
-    if len(sys.argv) >= 6:
+    if args.need_layers_name:
         outputs = args.need_layers_name.split(',')
 
     ret = MyNet.convert(npy_weight, fluid_model, outputs)
-- 
GitLab