From bc7751915a8c9b269f293c96520a86ac07fc2a36 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 4 May 2018 15:58:00 +0800
Subject: [PATCH] Fix mace tools enable openmp arg parse bug.

---
 tools/mace_tools.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/mace_tools.py b/tools/mace_tools.py
index 347f2f75..5d190704 100644
--- a/tools/mace_tools.py
+++ b/tools/mace_tools.py
@@ -386,8 +386,8 @@ def parse_args():
         help="Enable out of range check for opencl.")
     parser.add_argument(
         "--enable_openmp",
-        type="bool",
-        default="true",
+        type=str2bool,
+        default=True,
         help="Enable openmp.")
     parser.add_argument(
         "--omp_num_threads",
-- 
GitLab