summaryrefslogtreecommitdiffstats
path: root/scripts/global_setting.py
diff options
context:
space:
mode:
authorwangy122 <wangy122@chinatelecom.cn>2021-03-05 15:11:17 +0800
committerwangy122 <wangy122@chinatelecom.cn>2021-03-05 15:11:49 +0800
commitf15dc37e62eee9d5d02bf58af1053750a20fad23 (patch)
tree5599e6ab03df27a287fcb799b5a186c40dbaffe6 /scripts/global_setting.py
parentab00dc61c4f1206fbd906251b4c01ede0b89e990 (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.py36
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)
+
+