diff options
author | wangy122 <wangy122@chinatelecom.cn> | 2021-03-05 15:11:17 +0800 |
---|---|---|
committer | wangy122 <wangy122@chinatelecom.cn> | 2021-03-05 15:11:49 +0800 |
commit | f15dc37e62eee9d5d02bf58af1053750a20fad23 (patch) | |
tree | 5599e6ab03df27a287fcb799b5a186c40dbaffe6 /scripts/global_setting.py | |
parent | ab00dc61c4f1206fbd906251b4c01ede0b89e990 (diff) |
feat:add dockerfile
Issue-ID: USECASEUI-525
Signed-off-by: wangy122 <wangy122@chinatelecom.cn>
Change-Id: Ifca8abdfff479216bb0ea6b84d2c61fb640039f5
Diffstat (limited to 'scripts/global_setting.py')
-rw-r--r-- | scripts/global_setting.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/scripts/global_setting.py b/scripts/global_setting.py new file mode 100644 index 0000000..bb035f9 --- /dev/null +++ b/scripts/global_setting.py @@ -0,0 +1,36 @@ +from __future__ import absolute_import +from __future__ import division +from __future__ import print_function + + +import collections +import math +import modeling +import optimization +import tokenization +import six +import tensorflow as tf +import os + + + + +### Global variables + +# GPU number, default: -1, means not used +CUDA_VISIBLE_DEVICES="2" + +# Questions to be trained/predicted +questions = ['Communication Service Name','Max Number of UEs','Data Rate Downlink','Latency','Data Rate Uplink','Resource Sharing Level','Mobility','Area'] + +# Configuration file +FLAGS_bert_config_file = '/home/run/chinese_L-12_H-768_A-12/bert_config.json' +FLAGS_vocab_file = '/home/run/chinese_L-12_H-768_A-12/vocab.txt' +FLAGS_init_checkpoint_squad = '/home/run/chinese_L-12_H-768_A-12/bert_model.ckpt' + +max_seq_length = 512 + + +tokenizer_ch = tokenization.FullTokenizer(vocab_file=FLAGS_vocab_file, do_lower_case=True) + + |