From 86adbe420534d3d4c658c30e3714a67f6eacc83c Mon Sep 17 00:00:00 2001
From: macan <ma_cancan@163.com>
Date: Fri, 31 May 2019 10:13:20 +0800
Subject: [PATCH] update readme and make run.py

---
 README.md |  2 ++
 run.py    | 28 +++++++++++++++++-----------
 2 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/README.md b/README.md
index bcd5774..21f2818 100644
--- a/README.md
+++ b/README.md
@@ -32,6 +32,8 @@ cd BERT-BiLSTM-CRF-NER/
 python3 setup.py install
 ```
 
+if you do not want to install, you just need clone this project and reference the file of <run.py> to train the model or start the service. 
+
 ## UPDATE:
 - 2019.2.25 Fix some bug for ner service
 - 2019.2.19: add text classification service
diff --git a/run.py b/run.py
index 9a7e084..9f70eb3 100644
--- a/run.py
+++ b/run.py
@@ -13,9 +13,6 @@
 from __future__ import print_function
 
 
-# sys.path.append('.')
-
-
 def start_server():
     from bert_base.server import BertServer
     from bert_base.server.helper import get_run_args
@@ -27,15 +24,24 @@ def start_server():
     server.join()
 
 
-def start_client():
-    pass
+def train_ner():
+    import os
+    from bert_base.train.train_helper import get_args_parser
+    from bert_base.train.bert_lstm_ner import train
 
-
-# def train_ner():
-#     from bert_lstm_ner import main
-#     args =
-#     main()
+    args = get_args_parser()
+    if True:
+        import sys
+        param_str = '\n'.join(['%20s = %s' % (k, v) for k, v in sorted(vars(args).items())])
+        print('usage: %s\n%20s   %s\n%s\n%s\n' % (' '.join(sys.argv), 'ARG', 'VALUE', '_' * 50, param_str))
+    print(args)
+    os.environ['CUDA_VISIBLE_DEVICES'] = args.device_map
+    train(args=args)
 
 
 if __name__ == '__main__':
-    start_server()
\ No newline at end of file
+    """
+    如果想训练，那么直接 指定参数跑，如果想启动服务，那么注释掉train,打开server即可
+    """
+    train_ner()
+    #start_server()
\ No newline at end of file