vumichien commited on
Commit
4245809
·
1 Parent(s): 6e405a0

update model v1_0_7_8

Browse files
config.py CHANGED
@@ -22,20 +22,20 @@ STANDARD_NAME_MAP_DATA_FILE = os.path.join(DATA_DIR, "standardData.csv")
22
  HALF = "float16"
23
  MODEL_TYPE = None
24
  DEVICE_TYPE = "cpu"
25
- MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v_1_0_6"
26
  if HALF == "float16":
27
  SENTENCE_EMBEDDING_FILE = os.path.join(
28
  DATA_DIR,
29
- "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6_float16.pkl",
30
  )
31
  elif HALF == "bfloat16":
32
  SENTENCE_EMBEDDING_FILE = os.path.join(
33
  DATA_DIR,
34
- "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6_bfloat16.pkl",
35
  )
36
  else:
37
  SENTENCE_EMBEDDING_FILE = os.path.join(
38
  DATA_DIR,
39
- "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6.pkl",
40
  )
41
  # SENTENCE_EMBEDDING_FILE = None
 
22
  HALF = "float16"
23
  MODEL_TYPE = None
24
  DEVICE_TYPE = "cpu"
25
+ MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v1_0_7_8"
26
  if HALF == "float16":
27
  SENTENCE_EMBEDDING_FILE = os.path.join(
28
  DATA_DIR,
29
+ "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v1_0_7_8_float16.pkl",
30
  )
31
  elif HALF == "bfloat16":
32
  SENTENCE_EMBEDDING_FILE = os.path.join(
33
  DATA_DIR,
34
+ "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v1_0_7_8_bfloat16.pkl",
35
  )
36
  else:
37
  SENTENCE_EMBEDDING_FILE = os.path.join(
38
  DATA_DIR,
39
+ "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v1_0_7_8.pkl",
40
  )
41
  # SENTENCE_EMBEDDING_FILE = None
data/abstractData.csv ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 標準科目,基準項目名,標準項目名
2
+ 左官,ALC面,ALC面
3
+ 左官,OAフロア下地,OAフロア下地
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
+ 左官,表面強化材面,表面強化材面
48
+ 左官,表面保護材下地,表面保護材下地
49
+ 左官,浮床下地,浮床下地
50
+ 左官,防水下地,防水下地
51
+ 左官,磨き仕上げ,磨き仕上げ
52
+ 左官,木軸床組・OAフロア・鋼製床組下,
53
+ 左官,木床下地,木床組下地
54
+ 左官,木床組下地,木床組下地
55
+ 左官,湧水マット下,湧水マット下
56
+ 左官,家具下,家具下
57
+ 左官,畳下,畳下
58
+ 塗装,ALC面,ALC面
59
+ 塗装,ECP面,ECP面
60
+ 塗装,PC面,PC面
61
+ 塗装,SGP 溶融亜鉛メッキ面,溶融亜鉛メッキ面
62
+ 塗装,SGP溶融亜鉛メッキ面,溶融亜鉛メッキ面
63
+ 塗装,ガラスクロス面,ガラスクロス面
64
+ 塗装,クロス面,クロス面
65
+ 塗装,ケイカル板面,ケイカル板面
66
+ 塗装,ケイカル面,ケイカル面
67
+ 塗装,コンクリート面,コンクリート面
68
+ 塗装,シャッター面,シャッター面
69
+ 塗装,ステンレス面,ステンレス面
70
+ 塗装,デッキ面,デッキ面
71
+ 塗装,デッキ面 溶融亜鉛メッキ面,溶融亜鉛メッキ面
72
+ 塗装,デッキ面溶融亜鉛メッキ面,溶融亜鉛メッキ面
73
+ 塗装,バックボード裏,バックボード裏
74
+ 塗装,フローリング面,フローリング面
75
+ 塗装,ボード面,ボード面
76
+ 塗装,メラミン不燃化粧板面,メラミン不燃化粧板面
77
+ 塗装,モルタル面,モルタル面
78
+ 塗装,ロックウール吸音板面,ロックウール板面
79
+ 塗装,ロックウール板面,ロックウール板面
80
+ 塗装,亜鉛メッキ鋼面,溶融亜鉛メッキ面
81
+ 塗装,亜鉛メッキ面,亜鉛メッキ面
82
+ 塗装,亜鉛メッキ面,溶融亜鉛メッキ面
83
+ 塗装,塩化ビニル面,塩化ビニル面
84
+ 塗装,押出成形セメント板面,ECP面
85
+ 塗装,外装鋼板面,外装鋼板面
86
+ 塗装,既製コンクリート面,既製コンクリート面
87
+ 塗装,吸音板面,吸音板面
88
+ 塗装,鋼製建具面,鋼製建具面
89
+ 塗装,鋼板面,鉄部
90
+ 塗装,合板面,合板面
91
+ 塗装,人工木リブパネル面,木部
92
+ 塗装,人工木材部,木部
93
+ 塗装,人工木面,木部
94
+ 塗装,耐火ケイカル板面,耐火ケイカル板面
95
+ 塗装,耐火塗料面,耐火塗料面
96
+ 塗装,耐火被覆材面,耐火被覆材面
97
+ 塗装,鉄鋼,鉄部
98
+ 塗装,鉄鋼面,鉄部
99
+ 塗装,鉄鋼面 溶融亜鉛メッキ面,溶融亜鉛メッキ面
100
+ 塗装,鉄鋼面・溶融亜鉛メッキ面,溶融亜鉛メッキ面
101
+ 塗装,鉄鋼面溶融亜鉛メッキ面,溶融亜鉛メッキ面
102
+ 塗装,鉄骨 亜鉛メッキ面,溶融亜鉛メッキ面
103
+ 塗装,鉄骨 溶融亜鉛メッキ面,溶融亜鉛メッキ面
104
+ 塗装,鉄骨亜鉛メッキ面,溶融亜鉛メッキ面
105
+ 塗装,鉄骨面,鉄骨面
106
+ 塗装,鉄部,鉄部
107
+ 塗装,鉄部 亜鉛メッキ面,溶融亜鉛メッキ面
108
+ 塗装,鉄部 溶融亜鉛メッキ面,溶融亜鉛メッキ面
109
+ 塗装,鉄部溶融亜鉛メッキ面,溶融亜鉛メッキ面
110
+ 塗装,鉄面,鉄部
111
+ 塗装,木製建具面,木製建具面
112
+ 塗装,木部,���部
113
+ 塗装,木毛セメント板面,木毛セメント板面
114
+ 塗装,溶融亜鉛メッキ面,溶融亜鉛メッキ面
115
+ 塗装,練付合板面,木部
116
+ 塗装,塗床面,塗床面
117
+ コンクリート,捨てコンクリート,捨てコンクリート
118
+ コンクリート,免震層下部,免震層下部
119
+ コンクリート,機械基礎,機械基礎
120
+ コンクリート,嵩上げ,嵩上げ
121
+ コンクリート,防水保護,防水保護
122
+ コンクリート,捨てコンクリート,捨てコンクリート
123
+ コンクリート,基礎部マスコンクリート,基礎部マスコンクリート
124
+ コンクリート,基礎部,基礎部
125
+ コンクリート,免震装置下部,免震装置下部
126
+ コンクリート,免震層下部,免震層下部
127
+ コンクリート,地上部,地上部
128
+ コンクリート,機械基礎,機械基礎
129
+ コンクリート,嵩上げ,嵩上げ
130
+ コンクリート,防水保護,防水保護
131
+ コンクリート,湧水マット保護,湧水マット保護
132
+ コンクリート,EXP.J面,EXP.J面
133
+ コンクリート,土間コンクリート,土間コンクリート
data/anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v1_0_7_8_float16.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36ace71db8b7a92d096be06d7911d6c7fcb88de43d4fbfce6353765032bc09e0
3
+ size 11361956
data/standardData.csv CHANGED
The diff for this file is too large to render. See raw diff
 
prepare.py CHANGED
@@ -4,7 +4,7 @@ from sentence_transformers import (
4
  export_dynamic_quantized_onnx_model,
5
  )
6
 
7
- MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v_1_0_6"
8
 
9
 
10
  def export_model(backend="onnx", use_qint8=False):
@@ -33,10 +33,10 @@ def export_model(backend="onnx", use_qint8=False):
33
 
34
 
35
  # Export all combinations
36
- # backend = "onnx"
37
- # use_qint8 = False
38
- # print(f"Exporting {backend} model with QINT8={use_qint8}")
39
- # export_model(backend=backend, use_qint8=use_qint8)
40
 
41
  # import tensorrt as trt
42
  # print(trt.__version__)
 
4
  export_dynamic_quantized_onnx_model,
5
  )
6
 
7
+ MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v1_0_7_8"
8
 
9
 
10
  def export_model(backend="onnx", use_qint8=False):
 
33
 
34
 
35
  # Export all combinations
36
+ BACKEND = "openvino"
37
+ USE_QINT8 = False
38
+ print(f"Exporting {BACKEND} model with QINT8={USE_QINT8}")
39
+ export_model(backend=BACKEND, use_qint8=USE_QINT8)
40
 
41
  # import tensorrt as trt
42
  # print(trt.__version__)