diff --git a/bert_base/train/train_helper.py b/bert_base/train/train_helper.py
index 4b95be5..3484c3e 100644
--- a/bert_base/train/train_helper.py
+++ b/bert_base/train/train_helper.py
@@ -37,12 +37,12 @@ def get_args_parser():
     group2 = parser.add_argument_group('Model Config', 'config the model params')
     group2.add_argument('-max_seq_length', type=int, default=128,
                         help='The maximum total input sequence length after WordPiece tokenization.')
-    group2.add_argument('-do_train', type=bool, default=True,
+    group2.add_argument('-do_train', action='store_false', default=True,
                         help='Whether to run training.')
-    group2.add_argument('-do_eval', type=bool, default=True,
+    group2.add_argument('-do_eval', action='store_false', default=True,
                         help='Whether to run eval on the dev set.')
-    group2.add_argument('-do_predict', type=bool, default=True,
-                        help='Whether to run the model in inference mode on the test set.')
+    group2.add_argument('-do_predict', action='store_false', default=True,
+                        help='Whether to run the predict in inference mode on the test set.')
     group2.add_argument('-batch_size', type=int, default=64,
                         help='Total batch size for training, eval and predict.')
     group2.add_argument('-learning_rate', type=float, default=1e-5,
diff --git a/setup.py b/setup.py
index 7d45c1d..396107c 100644
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,5 @@
+# encoding =utf-8
+
 from os import path
 import codecs
 from setuptools import setup, find_packages
@@ -11,7 +13,7 @@
 # print(__version__)
 setup(
     name='bert_base',
-    version='0.0.8',
+    version='0.0.9',
     description='Use Google\'s BERT for Chinese natural language processing tasks such as named entity recognition and provide server services',
     url='https://github.com/macanv/BERT-BiLSTM-CRF-NER',
     long_description=open('README.md', 'r', encoding='utf-8').read(),