forked from macanv/BERT-BiLSTM-CRF-NER
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.py
47 lines (37 loc) · 1.18 KB
/
run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
运行 BERT NER Server
#@Time : 2019/1/26 21:00
# @Author : MaCan ([email protected])
# @File : run.py
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
def start_server():
from bert_base.server import BertServer
from bert_base.server.helper import get_run_args
args = get_run_args()
print(args)
server = BertServer(args)
server.start()
server.join()
def train_ner():
import os
from bert_base.train.train_helper import get_args_parser
from bert_base.train.bert_lstm_ner import train
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__':
"""
如果想训练,那么直接 指定参数跑,如果想启动服务,那么注释掉train,打开server即可
"""
train_ner()
#start_server()