qinghuiwan commited on
Commit
ecaf2df
·
verified ·
1 Parent(s): b464ac5

Upload folder using huggingface_hub

Browse files
Files changed (43) hide show
  1. 1_Pooling/config.json +10 -0
  2. README.md +497 -0
  3. checkpoint-4167/1_Pooling/config.json +10 -0
  4. checkpoint-4167/README.md +484 -0
  5. checkpoint-4167/config.json +31 -0
  6. checkpoint-4167/config_sentence_transformers.json +14 -0
  7. checkpoint-4167/model.safetensors +3 -0
  8. checkpoint-4167/modules.json +14 -0
  9. checkpoint-4167/optimizer.pt +3 -0
  10. checkpoint-4167/rng_state.pth +3 -0
  11. checkpoint-4167/scheduler.pt +3 -0
  12. checkpoint-4167/sentence_bert_config.json +4 -0
  13. checkpoint-4167/special_tokens_map.json +7 -0
  14. checkpoint-4167/tokenizer.json +0 -0
  15. checkpoint-4167/tokenizer_config.json +58 -0
  16. checkpoint-4167/trainer_state.json +687 -0
  17. checkpoint-4167/training_args.bin +3 -0
  18. checkpoint-4167/vocab.txt +0 -0
  19. checkpoint-4630/1_Pooling/config.json +10 -0
  20. checkpoint-4630/README.md +497 -0
  21. checkpoint-4630/config.json +31 -0
  22. checkpoint-4630/config_sentence_transformers.json +14 -0
  23. checkpoint-4630/model.safetensors +3 -0
  24. checkpoint-4630/modules.json +14 -0
  25. checkpoint-4630/optimizer.pt +3 -0
  26. checkpoint-4630/rng_state.pth +3 -0
  27. checkpoint-4630/scheduler.pt +3 -0
  28. checkpoint-4630/sentence_bert_config.json +4 -0
  29. checkpoint-4630/special_tokens_map.json +7 -0
  30. checkpoint-4630/tokenizer.json +0 -0
  31. checkpoint-4630/tokenizer_config.json +58 -0
  32. checkpoint-4630/trainer_state.json +758 -0
  33. checkpoint-4630/training_args.bin +3 -0
  34. checkpoint-4630/vocab.txt +0 -0
  35. config.json +31 -0
  36. config_sentence_transformers.json +14 -0
  37. model.safetensors +3 -0
  38. modules.json +14 -0
  39. sentence_bert_config.json +4 -0
  40. special_tokens_map.json +7 -0
  41. tokenizer.json +0 -0
  42. tokenizer_config.json +58 -0
  43. vocab.txt +0 -0
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
README.md ADDED
@@ -0,0 +1,497 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:7400
9
+ - loss:MultipleNegativesRankingLoss
10
+ base_model: shibing624/text2vec-base-chinese
11
+ widget:
12
+ - source_sentence: 在一条无限长的均匀管道中,流体在任何截面的行为都一样。这种沿管道方向的不变性直接保证了流量在管道中处处相等。正是因为没有哪个位置特殊,流体才不会在某处堆积或消散。
13
+ sentences:
14
+ - 一条法律规则如果对所有人一视同仁——不因身份、地位而区别对待——那么在这条规则下,所有人的权利总量保持均衡。平等对待的对称性直接导致权利分配的守恒。
15
+ - 一个极少地震的地区,地震仪几乎永远显示平静,每天的读数不传递新信息。但在地震活跃带,每次仪器读数都可能是不同强度和方向的震动,每条数据都在揭示地壳应力的新变化,监测价值远高于前者。
16
+ - 下棋时,你每一步都假设对手会走对他最有利的着法,然后在这种假设下选择对自己最有利的走法。对手也这样想。双方都做最坏打算后的最优选择,最终棋局会走向一个双方都无法改善的局面。
17
+ - source_sentence: 一条河流遭受污染后,下游的污染物浓度逐渐降低。最初几公里下降很快,但之后稀释速度放缓,残余污染在很长的河段上维持着低但可检测的浓度,远比预期消散得更慢。
18
+ sentences:
19
+ - 新修一条地铁线路后,沿线房价和人口不会立刻变化。三五年后居民逐渐迁入,商业配套慢慢成熟,片区终于繁荣起来。但此时交通压力又超出设计承载力,需要再次扩建,而扩建的效果又要等很多年才显现。
20
+ - 知识从教师集中的城市向师资匮乏的乡村扩散——通过支教、在线课程、教材传播。知识资源丰富的地方像高浓度溶液,自然向匮乏之处渗透,差距越大渗透的动力就越强。
21
+ - 一只股票在利空消息后暴跌,第一天跌幅最大。之后几天继续下跌但幅度递减,随后进入漫长的阴跌期——每天只跌一点点,但持续很久。这种缓慢出血比暴跌更让投资者煎熬。
22
+ - source_sentence: 今天的气温很大程度上取决于昨天和前天的气温,再加上一些随机扰动——比如一阵突然的冷风。天气预报之所以短期比较准、长期越来越不靠谱,就是因为每天的天气都是前几天天气的延续加上不可预测的新干扰的叠加。
23
+ sentences:
24
+ - 所有博主都在晚上八点发视频,因为这是流量高峰。如果你换到其他时段,观众更少;如果大家都在八点发,竞争虽然激烈但流量也最大。每个博主都没有动力单独改变发布时间。
25
+ - 脑电图记录了头皮表面的微弱电信号。把这个复杂信号拆解成频率成分后,发现不同状态下主导频率不同:清醒专注时以中高频为主,放松时以中低频为主,深睡时以极低频为主。大脑的活动状态可以通过频率成分的变化来判读。
26
+ - 今年的粮食产量与前几年有关——连续丰年后土壤肥力可能下降,连续歉收后农民会调整种植策略。每年的天气条件是不可预测的随机因素。农业产量的年际变化就是历史趋势的延续叠加上气候随机性的结果。
27
+ - source_sentence: 某些金属合金在高温下柔软易弯。当温度降到某个值时,内部结构突然重新排列,金属瞬间变得坚硬而脆。升温时恢复柔软的温度却比冷却时变硬的温度高出不少,两个转变温度之间存在明显的不对称。
28
+ sentences:
29
+ - 有机化学家要从简单原料合成目标药物分子,每一步反应都有不同的产率和成本。在所有可能的合成路线中,他们要找到步骤最少、总产率最高的那条反应路径。
30
+ - 热钱在全球市场中有一个主流方向——从低利率国家流向高利率国家。但每一笔具体的资金流向又带有投资者个体判断的随机性。宏观的利差驱动和微观的个体分散叠加,形成了资本流动的全貌。
31
+ - 暖湿气流和冷空气对峙时,天气在阴沉和放晴之间维持微妙的平衡。当冷空气势力刚好越过某个强度,锋面突然南压,天气从温暖直接跳变为寒潮。锋面过后气温回升的过程却是缓慢渐进的,冷暖转换不对等。
32
+ - source_sentence: 肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。
33
+ sentences:
34
+ - 在一个人人收入相同的社会,你知道一个人的收入不会获得任何关于他身份的信息。但在收入差距大且分布均匀的社会,每了解一个人的收入都能帮你推断他的职业、教育和生活方式,信息量丰富得多。
35
+ - 国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。
36
+ - 金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。
37
+ pipeline_tag: sentence-similarity
38
+ library_name: sentence-transformers
39
+ ---
40
+
41
+ # SentenceTransformer based on shibing624/text2vec-base-chinese
42
+
43
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese). It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
44
+
45
+ ## Model Details
46
+
47
+ ### Model Description
48
+ - **Model Type:** Sentence Transformer
49
+ - **Base model:** [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese) <!-- at revision 183bb99aa7af74355fb58d16edf8c13ae7c5433e -->
50
+ - **Maximum Sequence Length:** 128 tokens
51
+ - **Output Dimensionality:** 768 dimensions
52
+ - **Similarity Function:** Cosine Similarity
53
+ <!-- - **Training Dataset:** Unknown -->
54
+ <!-- - **Language:** Unknown -->
55
+ <!-- - **License:** Unknown -->
56
+
57
+ ### Model Sources
58
+
59
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
60
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
61
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
62
+
63
+ ### Full Model Architecture
64
+
65
+ ```
66
+ SentenceTransformer(
67
+ (0): Transformer({'max_seq_length': 128, 'do_lower_case': False, 'architecture': 'BertModel'})
68
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
69
+ )
70
+ ```
71
+
72
+ ## Usage
73
+
74
+ ### Direct Usage (Sentence Transformers)
75
+
76
+ First install the Sentence Transformers library:
77
+
78
+ ```bash
79
+ pip install -U sentence-transformers
80
+ ```
81
+
82
+ Then you can load this model and run inference.
83
+ ```python
84
+ from sentence_transformers import SentenceTransformer
85
+
86
+ # Download from the 🤗 Hub
87
+ model = SentenceTransformer("sentence_transformers_model_id")
88
+ # Run inference
89
+ sentences = [
90
+ '肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。',
91
+ '金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。',
92
+ '国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。',
93
+ ]
94
+ embeddings = model.encode(sentences)
95
+ print(embeddings.shape)
96
+ # [3, 768]
97
+
98
+ # Get the similarity scores for the embeddings
99
+ similarities = model.similarity(embeddings, embeddings)
100
+ print(similarities)
101
+ # tensor([[1.0000, 0.9134, 0.0324],
102
+ # [0.9134, 1.0000, 0.1733],
103
+ # [0.0324, 0.1733, 1.0000]])
104
+ ```
105
+
106
+ <!--
107
+ ### Direct Usage (Transformers)
108
+
109
+ <details><summary>Click to see the direct usage in Transformers</summary>
110
+
111
+ </details>
112
+ -->
113
+
114
+ <!--
115
+ ### Downstream Usage (Sentence Transformers)
116
+
117
+ You can finetune this model on your own dataset.
118
+
119
+ <details><summary>Click to expand</summary>
120
+
121
+ </details>
122
+ -->
123
+
124
+ <!--
125
+ ### Out-of-Scope Use
126
+
127
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
128
+ -->
129
+
130
+ <!--
131
+ ## Bias, Risks and Limitations
132
+
133
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
134
+ -->
135
+
136
+ <!--
137
+ ### Recommendations
138
+
139
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
140
+ -->
141
+
142
+ ## Training Details
143
+
144
+ ### Training Dataset
145
+
146
+ #### Unnamed Dataset
147
+
148
+ * Size: 7,400 training samples
149
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
150
+ * Approximate statistics based on the first 1000 samples:
151
+ | | sentence1 | sentence2 | label |
152
+ |:--------|:-----------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
153
+ | type | string | string | float |
154
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.8 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.35 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
155
+ * Samples:
156
+ | sentence1 | sentence2 | label |
157
+ |:-----------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------|:-----------------|
158
+ | <code>急诊室里病人到达的时间没有任何规律。可能一个小时来六个人,下一个小时只来一个。但每个人来不来和其他人完全无关,平均下来每小时到达人数基本稳定。等待时间有时很短有时很长。</code> | <code>保险公司收到的理赔申请在时间上零散分布。本周三个下周可能七个,具体哪天收到完全不可预测。但年度总理赔数很稳定,每一起事故独立发生,一起理赔不影响下一起何时出现。</code> | <code>1.0</code> |
159
+ | <code>经济系统中几十个指标——GDP、失业率、通胀率——互相影响看似复杂。但分析后发现,它们的联动可以归结为几个独立的方向:一个方向代表经济周期,另一个方向代表结构调整。沿着这些方向看,经济的行为最简单。</code> | <code>股市里几千只股票每天的涨跌看似无序,但分析后发现大部分波动可以归结为几个主要方向——一个是大盘整体涨跌,一个是行业板块轮动。沿着这几个方向看,市场的行为变得最简单最可预测。</code> | <code>1.0</code> |
160
+ | <code>天气中既有持续几小时的雷暴这种短时高频事件,也有持续几天的冷锋这种中期过程,还有季节变化这种长期低频趋势。气象分析需要在不同时间尺度上同时观察——用小窗口看雷暴的精确时间,用大窗口看季节的整体趋势。</code> | <code>说话时辅音是短促的高频爆发持续几十毫秒,元音是持续较长的低频振动。语音识别需要同时精确定位短促辅音的出现时刻和长元音的频率特征。用短窗口分析辅音的时间细节,用长窗口分析元音的频率细节。</code> | <code>1.0</code> |
161
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
162
+ ```json
163
+ {
164
+ "scale": 20.0,
165
+ "similarity_fct": "cos_sim",
166
+ "gather_across_devices": false
167
+ }
168
+ ```
169
+
170
+ ### Evaluation Dataset
171
+
172
+ #### Unnamed Dataset
173
+
174
+ * Size: 823 evaluation samples
175
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
176
+ * Approximate statistics based on the first 823 samples:
177
+ | | sentence1 | sentence2 | label |
178
+ |:--------|:------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
179
+ | type | string | string | float |
180
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.98 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.22 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
181
+ * Samples:
182
+ | sentence1 | sentence2 | label |
183
+ |:----------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------|:-----------------|
184
+ | <code>一个人今天的情绪状态很大程度上延续昨天的情绪。心情好的惯性会持续几天,心情差也会低落一阵。但每天都可能遇到随机事件——收到好消息或坏消息——把情绪轨迹推离原来的方向。情绪的变化就是历史惯性加上每天的随机冲击。</code> | <code>一段文章中下一个词的选择很大程度上取决于前面几个词。说了因为就很可能接所以,说了虽然就很可能接但是。但作者偶尔会突破预期用一个意外的词制造效果。文本的生成像是对已有上下文的延续加上创造性的随机变化。</code> | <code>1.0</code> |
185
+ | <code>一个信号可以用随时间变化的波形来描述,也可以用各频率成分的强度来描述。这两种描述是同一信号的镜像刻画。在时间域里难以解决的问题转到频率域可能很简单,反之亦然。两种视角提供了看同一问题的对称方式。</code> | <code>国际贸易谈判中,出口国追求的高价恰好是进口国要避免的高价。出口方的最优策略和进口方的最优策略是同一个博弈的镜像刻画。谈判桌两边的分析虽然立场相反,但使用的逻辑框架完全对称。</code> | <code>1.0</code> |
186
+ | <code>一个社区从高度凝聚走向分散化的过程中,并不是某天突然崩解。在转折期,各种规模的社交圈子此起彼伏地形成又解散,小到三五个人大到上百人的聚集同时存在,群体行为的波动达到最大,各种规模的组织形态都能观察到。</code> | <code>某种合金在特定温度附近,微观结构表现出异常行为。原子排列的有序和无序区域在各个尺度上共存,从纳米到微米都能看到两种状态的混合。材料的比热容在这个温度附近出现异常尖峰,而且不同成分的合金在各自的临界温度附近表现出惊人的相似规律。</code> | <code>1.0</code> |
187
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
188
+ ```json
189
+ {
190
+ "scale": 20.0,
191
+ "similarity_fct": "cos_sim",
192
+ "gather_across_devices": false
193
+ }
194
+ ```
195
+
196
+ ### Training Hyperparameters
197
+ #### Non-Default Hyperparameters
198
+
199
+ - `eval_strategy`: epoch
200
+ - `per_device_train_batch_size`: 16
201
+ - `per_device_eval_batch_size`: 16
202
+ - `learning_rate`: 2e-05
203
+ - `num_train_epochs`: 10
204
+ - `warmup_ratio`: 0.1
205
+ - `use_mps_device`: True
206
+ - `dataloader_pin_memory`: False
207
+
208
+ #### All Hyperparameters
209
+ <details><summary>Click to expand</summary>
210
+
211
+ - `overwrite_output_dir`: False
212
+ - `do_predict`: False
213
+ - `eval_strategy`: epoch
214
+ - `prediction_loss_only`: True
215
+ - `per_device_train_batch_size`: 16
216
+ - `per_device_eval_batch_size`: 16
217
+ - `per_gpu_train_batch_size`: None
218
+ - `per_gpu_eval_batch_size`: None
219
+ - `gradient_accumulation_steps`: 1
220
+ - `eval_accumulation_steps`: None
221
+ - `torch_empty_cache_steps`: None
222
+ - `learning_rate`: 2e-05
223
+ - `weight_decay`: 0.0
224
+ - `adam_beta1`: 0.9
225
+ - `adam_beta2`: 0.999
226
+ - `adam_epsilon`: 1e-08
227
+ - `max_grad_norm`: 1.0
228
+ - `num_train_epochs`: 10
229
+ - `max_steps`: -1
230
+ - `lr_scheduler_type`: linear
231
+ - `lr_scheduler_kwargs`: None
232
+ - `warmup_ratio`: 0.1
233
+ - `warmup_steps`: 0
234
+ - `log_level`: passive
235
+ - `log_level_replica`: warning
236
+ - `log_on_each_node`: True
237
+ - `logging_nan_inf_filter`: True
238
+ - `save_safetensors`: True
239
+ - `save_on_each_node`: False
240
+ - `save_only_model`: False
241
+ - `restore_callback_states_from_checkpoint`: False
242
+ - `no_cuda`: False
243
+ - `use_cpu`: False
244
+ - `use_mps_device`: True
245
+ - `seed`: 42
246
+ - `data_seed`: None
247
+ - `jit_mode_eval`: False
248
+ - `bf16`: False
249
+ - `fp16`: False
250
+ - `fp16_opt_level`: O1
251
+ - `half_precision_backend`: auto
252
+ - `bf16_full_eval`: False
253
+ - `fp16_full_eval`: False
254
+ - `tf32`: None
255
+ - `local_rank`: 0
256
+ - `ddp_backend`: None
257
+ - `tpu_num_cores`: None
258
+ - `tpu_metrics_debug`: False
259
+ - `debug`: []
260
+ - `dataloader_drop_last`: False
261
+ - `dataloader_num_workers`: 0
262
+ - `dataloader_prefetch_factor`: None
263
+ - `past_index`: -1
264
+ - `disable_tqdm`: False
265
+ - `remove_unused_columns`: True
266
+ - `label_names`: None
267
+ - `load_best_model_at_end`: False
268
+ - `ignore_data_skip`: False
269
+ - `fsdp`: []
270
+ - `fsdp_min_num_params`: 0
271
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
272
+ - `fsdp_transformer_layer_cls_to_wrap`: None
273
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
274
+ - `parallelism_config`: None
275
+ - `deepspeed`: None
276
+ - `label_smoothing_factor`: 0.0
277
+ - `optim`: adamw_torch_fused
278
+ - `optim_args`: None
279
+ - `adafactor`: False
280
+ - `group_by_length`: False
281
+ - `length_column_name`: length
282
+ - `project`: huggingface
283
+ - `trackio_space_id`: trackio
284
+ - `ddp_find_unused_parameters`: None
285
+ - `ddp_bucket_cap_mb`: None
286
+ - `ddp_broadcast_buffers`: False
287
+ - `dataloader_pin_memory`: False
288
+ - `dataloader_persistent_workers`: False
289
+ - `skip_memory_metrics`: True
290
+ - `use_legacy_prediction_loop`: False
291
+ - `push_to_hub`: False
292
+ - `resume_from_checkpoint`: None
293
+ - `hub_model_id`: None
294
+ - `hub_strategy`: every_save
295
+ - `hub_private_repo`: None
296
+ - `hub_always_push`: False
297
+ - `hub_revision`: None
298
+ - `gradient_checkpointing`: False
299
+ - `gradient_checkpointing_kwargs`: None
300
+ - `include_inputs_for_metrics`: False
301
+ - `include_for_metrics`: []
302
+ - `eval_do_concat_batches`: True
303
+ - `fp16_backend`: auto
304
+ - `push_to_hub_model_id`: None
305
+ - `push_to_hub_organization`: None
306
+ - `mp_parameters`:
307
+ - `auto_find_batch_size`: False
308
+ - `full_determinism`: False
309
+ - `torchdynamo`: None
310
+ - `ray_scope`: last
311
+ - `ddp_timeout`: 1800
312
+ - `torch_compile`: False
313
+ - `torch_compile_backend`: None
314
+ - `torch_compile_mode`: None
315
+ - `include_tokens_per_second`: False
316
+ - `include_num_input_tokens_seen`: no
317
+ - `neftune_noise_alpha`: None
318
+ - `optim_target_modules`: None
319
+ - `batch_eval_metrics`: False
320
+ - `eval_on_start`: False
321
+ - `use_liger_kernel`: False
322
+ - `liger_kernel_config`: None
323
+ - `eval_use_gather_object`: False
324
+ - `average_tokens_across_devices`: True
325
+ - `prompts`: None
326
+ - `batch_sampler`: batch_sampler
327
+ - `multi_dataset_batch_sampler`: proportional
328
+ - `router_mapping`: {}
329
+ - `learning_rate_mapping`: {}
330
+
331
+ </details>
332
+
333
+ ### Training Logs
334
+ <details><summary>Click to expand</summary>
335
+
336
+ | Epoch | Step | Training Loss | Validation Loss |
337
+ |:------:|:----:|:-------------:|:---------------:|
338
+ | 0.1080 | 50 | 1.9486 | - |
339
+ | 0.2160 | 100 | 1.2593 | - |
340
+ | 0.3240 | 150 | 0.7355 | - |
341
+ | 0.4320 | 200 | 0.546 | - |
342
+ | 0.5400 | 250 | 0.4791 | - |
343
+ | 0.6479 | 300 | 0.3971 | - |
344
+ | 0.7559 | 350 | 0.3148 | - |
345
+ | 0.8639 | 400 | 0.3241 | - |
346
+ | 0.9719 | 450 | 0.2552 | - |
347
+ | 1.0 | 463 | - | 0.2700 |
348
+ | 1.0799 | 500 | 0.2515 | - |
349
+ | 1.1879 | 550 | 0.2607 | - |
350
+ | 1.2959 | 600 | 0.2093 | - |
351
+ | 1.4039 | 650 | 0.2755 | - |
352
+ | 1.5119 | 700 | 0.1779 | - |
353
+ | 1.6199 | 750 | 0.1862 | - |
354
+ | 1.7279 | 800 | 0.1645 | - |
355
+ | 1.8359 | 850 | 0.1676 | - |
356
+ | 1.9438 | 900 | 0.2098 | - |
357
+ | 2.0 | 926 | - | 0.1917 |
358
+ | 2.0518 | 950 | 0.2279 | - |
359
+ | 2.1598 | 1000 | 0.1753 | - |
360
+ | 2.2678 | 1050 | 0.181 | - |
361
+ | 2.3758 | 1100 | 0.1755 | - |
362
+ | 2.4838 | 1150 | 0.1633 | - |
363
+ | 2.5918 | 1200 | 0.13 | - |
364
+ | 2.6998 | 1250 | 0.1237 | - |
365
+ | 2.8078 | 1300 | 0.1911 | - |
366
+ | 2.9158 | 1350 | 0.1655 | - |
367
+ | 3.0 | 1389 | - | 0.1829 |
368
+ | 3.0238 | 1400 | 0.1548 | - |
369
+ | 3.1317 | 1450 | 0.1487 | - |
370
+ | 3.2397 | 1500 | 0.1692 | - |
371
+ | 3.3477 | 1550 | 0.1719 | - |
372
+ | 3.4557 | 1600 | 0.1541 | - |
373
+ | 3.5637 | 1650 | 0.1664 | - |
374
+ | 3.6717 | 1700 | 0.1451 | - |
375
+ | 3.7797 | 1750 | 0.163 | - |
376
+ | 3.8877 | 1800 | 0.1615 | - |
377
+ | 3.9957 | 1850 | 0.1445 | - |
378
+ | 4.0 | 1852 | - | 0.1837 |
379
+ | 4.1037 | 1900 | 0.1527 | - |
380
+ | 4.2117 | 1950 | 0.1439 | - |
381
+ | 4.3197 | 2000 | 0.1321 | - |
382
+ | 4.4276 | 2050 | 0.1593 | - |
383
+ | 4.5356 | 2100 | 0.1388 | - |
384
+ | 4.6436 | 2150 | 0.1434 | - |
385
+ | 4.7516 | 2200 | 0.1456 | - |
386
+ | 4.8596 | 2250 | 0.1568 | - |
387
+ | 4.9676 | 2300 | 0.1549 | - |
388
+ | 5.0 | 2315 | - | 0.1620 |
389
+ | 5.0756 | 2350 | 0.1736 | - |
390
+ | 5.1836 | 2400 | 0.1337 | - |
391
+ | 5.2916 | 2450 | 0.1221 | - |
392
+ | 5.3996 | 2500 | 0.1372 | - |
393
+ | 5.5076 | 2550 | 0.1682 | - |
394
+ | 5.6156 | 2600 | 0.1489 | - |
395
+ | 5.7235 | 2650 | 0.138 | - |
396
+ | 5.8315 | 2700 | 0.1197 | - |
397
+ | 5.9395 | 2750 | 0.139 | - |
398
+ | 6.0 | 2778 | - | 0.1673 |
399
+ | 6.0475 | 2800 | 0.1503 | - |
400
+ | 6.1555 | 2850 | 0.1441 | - |
401
+ | 6.2635 | 2900 | 0.1614 | - |
402
+ | 6.3715 | 2950 | 0.1179 | - |
403
+ | 6.4795 | 3000 | 0.1509 | - |
404
+ | 6.5875 | 3050 | 0.1467 | - |
405
+ | 6.6955 | 3100 | 0.141 | - |
406
+ | 6.8035 | 3150 | 0.1358 | - |
407
+ | 6.9114 | 3200 | 0.1639 | - |
408
+ | 7.0 | 3241 | - | 0.1586 |
409
+ | 7.0194 | 3250 | 0.1294 | - |
410
+ | 7.1274 | 3300 | 0.1344 | - |
411
+ | 7.2354 | 3350 | 0.1138 | - |
412
+ | 7.3434 | 3400 | 0.1409 | - |
413
+ | 7.4514 | 3450 | 0.1293 | - |
414
+ | 7.5594 | 3500 | 0.1088 | - |
415
+ | 7.6674 | 3550 | 0.1399 | - |
416
+ | 7.7754 | 3600 | 0.127 | - |
417
+ | 7.8834 | 3650 | 0.1448 | - |
418
+ | 7.9914 | 3700 | 0.1376 | - |
419
+ | 8.0 | 3704 | - | 0.1555 |
420
+ | 8.0994 | 3750 | 0.1171 | - |
421
+ | 8.2073 | 3800 | 0.1221 | - |
422
+ | 8.3153 | 3850 | 0.1406 | - |
423
+ | 8.4233 | 3900 | 0.1299 | - |
424
+ | 8.5313 | 3950 | 0.1363 | - |
425
+ | 8.6393 | 4000 | 0.117 | - |
426
+ | 8.7473 | 4050 | 0.1572 | - |
427
+ | 8.8553 | 4100 | 0.1213 | - |
428
+ | 8.9633 | 4150 | 0.1218 | - |
429
+ | 9.0 | 4167 | - | 0.1551 |
430
+ | 9.0713 | 4200 | 0.1333 | - |
431
+ | 9.1793 | 4250 | 0.1339 | - |
432
+ | 9.2873 | 4300 | 0.1477 | - |
433
+ | 9.3952 | 4350 | 0.1013 | - |
434
+ | 9.5032 | 4400 | 0.1241 | - |
435
+ | 9.6112 | 4450 | 0.1091 | - |
436
+ | 9.7192 | 4500 | 0.1077 | - |
437
+ | 9.8272 | 4550 | 0.1261 | - |
438
+ | 9.9352 | 4600 | 0.1396 | - |
439
+ | 10.0 | 4630 | - | 0.1545 |
440
+
441
+ </details>
442
+
443
+ ### Framework Versions
444
+ - Python: 3.9.6
445
+ - Sentence Transformers: 5.1.2
446
+ - Transformers: 4.57.6
447
+ - PyTorch: 2.8.0
448
+ - Accelerate: 1.10.1
449
+ - Datasets: 4.5.0
450
+ - Tokenizers: 0.22.2
451
+
452
+ ## Citation
453
+
454
+ ### BibTeX
455
+
456
+ #### Sentence Transformers
457
+ ```bibtex
458
+ @inproceedings{reimers-2019-sentence-bert,
459
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
460
+ author = "Reimers, Nils and Gurevych, Iryna",
461
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
462
+ month = "11",
463
+ year = "2019",
464
+ publisher = "Association for Computational Linguistics",
465
+ url = "https://arxiv.org/abs/1908.10084",
466
+ }
467
+ ```
468
+
469
+ #### MultipleNegativesRankingLoss
470
+ ```bibtex
471
+ @misc{henderson2017efficient,
472
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
473
+ author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
474
+ year={2017},
475
+ eprint={1705.00652},
476
+ archivePrefix={arXiv},
477
+ primaryClass={cs.CL}
478
+ }
479
+ ```
480
+
481
+ <!--
482
+ ## Glossary
483
+
484
+ *Clearly define terms in order to be accessible across audiences.*
485
+ -->
486
+
487
+ <!--
488
+ ## Model Card Authors
489
+
490
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
491
+ -->
492
+
493
+ <!--
494
+ ## Model Card Contact
495
+
496
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
497
+ -->
checkpoint-4167/1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
checkpoint-4167/README.md ADDED
@@ -0,0 +1,484 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:7400
9
+ - loss:MultipleNegativesRankingLoss
10
+ base_model: shibing624/text2vec-base-chinese
11
+ widget:
12
+ - source_sentence: 在一条无限长的均匀管道中,流体在任何截面的行为都一样。这种沿管道方向的不变性直接保证了流量在管道中处处相等。正是因为没有哪个位置特殊,流体才不会在某处堆积或消散。
13
+ sentences:
14
+ - 一条法律规则如果对所有人一视同仁——不因身份、地位而区别对待——那么在这条规则下,所有人的权利总量保持均衡。平等对待的对称性直接导致权利分配的守恒。
15
+ - 一个极少地震的地区,地震仪几乎永远显示平静,每天的读数不传递新信息。但在地震活跃带,每次仪器读数都可能是不同强度和方向的震动,每条数据都在揭示地壳应力的新变化,监测价值远高于前者。
16
+ - 下棋时,你每一步都假设对手会走对他最有利的着法,然后在这种假设下选择对自己最有利的走法。对手也这样想。双方都做最坏打算后的最优选择,最终棋局会走向一个双方都无法改善的局面。
17
+ - source_sentence: 一条河流遭受污染后,下游的污染物浓度逐渐降低。最初几公里下降很快,但之后稀释速度放缓,残余污染在很长的河段上维持着低但可检测的浓度,远比预期消散得更慢。
18
+ sentences:
19
+ - 新修一条地铁线路后,沿线房价和人口不会立刻变化。三五年后居民逐渐迁入,商业配套慢慢成熟,片区终于繁荣起来。但此时交通压力又超出设计承载力,需要再次扩建,而扩建的效果又要等很多年才显现。
20
+ - 知识从教师集中的城市向师资匮乏的乡村扩散——通过支教、在线课程、教材传播。知识资源丰富的地方像高浓度溶液,自然向匮乏之处渗透,差距越大渗透的动力就越强。
21
+ - 一只股票在利空消息后暴跌,第一天跌幅最大。之后几天继续下跌但幅度递减,随后进入漫长的阴跌期——每天只跌一点点,但持续很久。这种缓慢出血比暴跌更让投资者煎熬。
22
+ - source_sentence: 今天的气温很大程度上取决于昨天和前天的气温,再加上一些随机扰动——比如一阵突然的冷风。天气预报之所以短期比较准、长期越来越不靠谱,就是因为每天的天气都是前几天天气的延续加上不可预测的新干扰的叠加。
23
+ sentences:
24
+ - 所有博主都在晚上八点发视频,因为这是流量高峰。如果你换到其他时段,观众更少;如果大家都在八点发,竞争虽然激烈但流量也最大。每个博主都没有动力单独改变发布时间。
25
+ - 脑电图记录了头皮表面的微弱电信号。把这个复杂信号拆解成频率成分后,发现不同状态下主导频率不同:清醒专注时以中高频为主,放松时以中低频为主,深睡时以极低频为主。大脑的活动状态可以通过频率成分的变化来判读。
26
+ - 今年的粮食产量与前几年有关——连续丰年后土壤肥力可能下降,连续歉收后农民会调整种植策略。每年的天气条件是不可预测的随机因素。农业产量的年际变化就是历史趋势的延续叠加上气候随机性的结果。
27
+ - source_sentence: 某些金属合金在高温下柔软易弯。当温度降到某个值时,内部结构突然重新排列,金属瞬间变得坚硬而脆。升温时恢复柔软的温度却比冷却时变硬的温度高出不少,两个转变温度之间存在明显的不对称。
28
+ sentences:
29
+ - 有机化学家要从简单原料合成目标药物分子,每一步反应都有不同的产率和成本。在所有可能的合成路线中,他们要找到步骤最少、总产率最高的那条反应路径。
30
+ - 热钱在全球市场中有一个主流方向——从低利率国家流向高利率国家。但每一笔具体的资金流向又带有投资者个体判断的随机性。宏观的利差驱动和微观的个体分散叠加,形成了资本流动的全貌。
31
+ - 暖湿气流和冷空气对峙时,天气在阴沉和放晴之间维持微妙的平衡。当冷空气势力刚好越过某个强度,锋面突然南压,天气从温暖直接跳变为寒潮。锋面过后气温回升的过程却是缓慢渐进的,冷暖转换不对等。
32
+ - source_sentence: 肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。
33
+ sentences:
34
+ - 在一个人人收入相同的社会,你知道一个人的收入不会获得任何关于他身份的信息。但在收入差距大且分布均匀的社会,每了解一个人的收入都能帮你推断他的职业、教育和生活方式,信息量丰富得多。
35
+ - 国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。
36
+ - 金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。
37
+ pipeline_tag: sentence-similarity
38
+ library_name: sentence-transformers
39
+ ---
40
+
41
+ # SentenceTransformer based on shibing624/text2vec-base-chinese
42
+
43
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese). It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
44
+
45
+ ## Model Details
46
+
47
+ ### Model Description
48
+ - **Model Type:** Sentence Transformer
49
+ - **Base model:** [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese) <!-- at revision 183bb99aa7af74355fb58d16edf8c13ae7c5433e -->
50
+ - **Maximum Sequence Length:** 128 tokens
51
+ - **Output Dimensionality:** 768 dimensions
52
+ - **Similarity Function:** Cosine Similarity
53
+ <!-- - **Training Dataset:** Unknown -->
54
+ <!-- - **Language:** Unknown -->
55
+ <!-- - **License:** Unknown -->
56
+
57
+ ### Model Sources
58
+
59
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
60
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
61
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
62
+
63
+ ### Full Model Architecture
64
+
65
+ ```
66
+ SentenceTransformer(
67
+ (0): Transformer({'max_seq_length': 128, 'do_lower_case': False, 'architecture': 'BertModel'})
68
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
69
+ )
70
+ ```
71
+
72
+ ## Usage
73
+
74
+ ### Direct Usage (Sentence Transformers)
75
+
76
+ First install the Sentence Transformers library:
77
+
78
+ ```bash
79
+ pip install -U sentence-transformers
80
+ ```
81
+
82
+ Then you can load this model and run inference.
83
+ ```python
84
+ from sentence_transformers import SentenceTransformer
85
+
86
+ # Download from the 🤗 Hub
87
+ model = SentenceTransformer("sentence_transformers_model_id")
88
+ # Run inference
89
+ sentences = [
90
+ '肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。',
91
+ '金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。',
92
+ '国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。',
93
+ ]
94
+ embeddings = model.encode(sentences)
95
+ print(embeddings.shape)
96
+ # [3, 768]
97
+
98
+ # Get the similarity scores for the embeddings
99
+ similarities = model.similarity(embeddings, embeddings)
100
+ print(similarities)
101
+ # tensor([[1.0000, 0.9132, 0.0188],
102
+ # [0.9132, 1.0000, 0.1628],
103
+ # [0.0188, 0.1628, 1.0000]])
104
+ ```
105
+
106
+ <!--
107
+ ### Direct Usage (Transformers)
108
+
109
+ <details><summary>Click to see the direct usage in Transformers</summary>
110
+
111
+ </details>
112
+ -->
113
+
114
+ <!--
115
+ ### Downstream Usage (Sentence Transformers)
116
+
117
+ You can finetune this model on your own dataset.
118
+
119
+ <details><summary>Click to expand</summary>
120
+
121
+ </details>
122
+ -->
123
+
124
+ <!--
125
+ ### Out-of-Scope Use
126
+
127
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
128
+ -->
129
+
130
+ <!--
131
+ ## Bias, Risks and Limitations
132
+
133
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
134
+ -->
135
+
136
+ <!--
137
+ ### Recommendations
138
+
139
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
140
+ -->
141
+
142
+ ## Training Details
143
+
144
+ ### Training Dataset
145
+
146
+ #### Unnamed Dataset
147
+
148
+ * Size: 7,400 training samples
149
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
150
+ * Approximate statistics based on the first 1000 samples:
151
+ | | sentence1 | sentence2 | label |
152
+ |:--------|:-----------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
153
+ | type | string | string | float |
154
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.8 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.35 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
155
+ * Samples:
156
+ | sentence1 | sentence2 | label |
157
+ |:-----------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------|:-----------------|
158
+ | <code>急诊室里病人到达的时间没有任何规律。可能一个小时来六个人,下一个小时只来一个。但每个人来不来和其他人完全无关,平均下来每小时到达人数基本稳定。等待时间有时很短有时很长。</code> | <code>保险公司收到的理赔申请在时间上零散分布。本周三个下周可能七个,具体哪天收到完全不可预测。但年度总理赔数很稳定,每一起事故独立发生,一起理赔不影响下一起何时出现。</code> | <code>1.0</code> |
159
+ | <code>经济系统中几十个指标——GDP、失业率、通胀率——互相影响看似复杂。但分析后发现,它们的联动可以归结为几个独立的方向:一个方向代表经济周期,另一个方向代表结构调整。沿着这些方向看,经济的行为最简单。</code> | <code>股市里几千只股票每天的涨跌看似无序,但分析后发现大部分波动可以归结为几个主要方向——一个是大盘整体涨跌,一个是行业板块轮动。沿着这几个方向看,市场的行为变得最简单最可预测。</code> | <code>1.0</code> |
160
+ | <code>天气中既有持续几小时的雷暴这种短时高频事件,也有持续几天的冷锋这种中期过程,还有季节变化这种长期低频趋势。气象分析需要在不同时间尺度上同时观察——用小窗口看雷暴的精确时间,用大窗口看季节的整体趋势。</code> | <code>说话时辅音是短促的高频爆发持续几十毫秒,元音是持续较长的低频振动。语音识别需要同时精确定位短促辅音的出现时刻和长元音的频率特征。用短窗口分析辅音的时间细节,用长窗口分析元音的频率细节。</code> | <code>1.0</code> |
161
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
162
+ ```json
163
+ {
164
+ "scale": 20.0,
165
+ "similarity_fct": "cos_sim",
166
+ "gather_across_devices": false
167
+ }
168
+ ```
169
+
170
+ ### Evaluation Dataset
171
+
172
+ #### Unnamed Dataset
173
+
174
+ * Size: 823 evaluation samples
175
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
176
+ * Approximate statistics based on the first 823 samples:
177
+ | | sentence1 | sentence2 | label |
178
+ |:--------|:------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
179
+ | type | string | string | float |
180
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.98 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.22 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
181
+ * Samples:
182
+ | sentence1 | sentence2 | label |
183
+ |:----------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------|:-----------------|
184
+ | <code>一个人今天的情绪状态很大程度上延续昨天的情绪。心情好的惯性会持续几天,心情差也会低落一阵。但每天都可能遇到随机事件——收到好消息或坏消息——把情绪轨迹推离原来的方向。情绪的变化就是历史惯性加上每天的随机冲击。</code> | <code>一段文章中下一个词的选择很大程度上取决于前面几个词。说了因为就很可能接所以,说了虽然就很可能接但是。但作者偶尔会突破预期用一个意外的词制造效果。文本的生成像是对已有上下文的延续加上创造性的随机变化。</code> | <code>1.0</code> |
185
+ | <code>一个信号可以用随时间变化的波形来描述,也可以用各频率成分的强度来描述。这两种描述是同一信号的镜像刻画。在时间域里难以解决的问题转到频率域可能很简单,反之亦然。两种视角提供了看同一问题的对称方式。</code> | <code>国际贸易谈判中,出口国追求的高价恰好是进口国要避免的高价。出口方的最优策略和进口方的最优策略是同一个博弈的镜像刻画。谈判桌两边的分析虽然立场相反,但使用的逻辑框架完全对称。</code> | <code>1.0</code> |
186
+ | <code>一个社区从高度凝聚走向分散化的过程中,并不是某天突然崩解。在转折期,各种规模的社交圈子此起彼伏地形成又解散,小到三五个人大到上百人的聚集同时存在,群体行为的波动达到最大,各种规模的组织形态都能观察到。</code> | <code>某种合金在特定温度附近,微观结构表现出异常行为。原子排列的有序和无序区域在各个尺度上共存,从纳米到微米都能看到两种状态的混合。材料的比热容在这个温度附近出现异常尖峰,而且不同成分的合金在各自的临界温度附近表现出惊人的相似规律。</code> | <code>1.0</code> |
187
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
188
+ ```json
189
+ {
190
+ "scale": 20.0,
191
+ "similarity_fct": "cos_sim",
192
+ "gather_across_devices": false
193
+ }
194
+ ```
195
+
196
+ ### Training Hyperparameters
197
+ #### Non-Default Hyperparameters
198
+
199
+ - `eval_strategy`: epoch
200
+ - `per_device_train_batch_size`: 16
201
+ - `per_device_eval_batch_size`: 16
202
+ - `learning_rate`: 2e-05
203
+ - `num_train_epochs`: 10
204
+ - `warmup_ratio`: 0.1
205
+ - `use_mps_device`: True
206
+ - `dataloader_pin_memory`: False
207
+
208
+ #### All Hyperparameters
209
+ <details><summary>Click to expand</summary>
210
+
211
+ - `overwrite_output_dir`: False
212
+ - `do_predict`: False
213
+ - `eval_strategy`: epoch
214
+ - `prediction_loss_only`: True
215
+ - `per_device_train_batch_size`: 16
216
+ - `per_device_eval_batch_size`: 16
217
+ - `per_gpu_train_batch_size`: None
218
+ - `per_gpu_eval_batch_size`: None
219
+ - `gradient_accumulation_steps`: 1
220
+ - `eval_accumulation_steps`: None
221
+ - `torch_empty_cache_steps`: None
222
+ - `learning_rate`: 2e-05
223
+ - `weight_decay`: 0.0
224
+ - `adam_beta1`: 0.9
225
+ - `adam_beta2`: 0.999
226
+ - `adam_epsilon`: 1e-08
227
+ - `max_grad_norm`: 1.0
228
+ - `num_train_epochs`: 10
229
+ - `max_steps`: -1
230
+ - `lr_scheduler_type`: linear
231
+ - `lr_scheduler_kwargs`: None
232
+ - `warmup_ratio`: 0.1
233
+ - `warmup_steps`: 0
234
+ - `log_level`: passive
235
+ - `log_level_replica`: warning
236
+ - `log_on_each_node`: True
237
+ - `logging_nan_inf_filter`: True
238
+ - `save_safetensors`: True
239
+ - `save_on_each_node`: False
240
+ - `save_only_model`: False
241
+ - `restore_callback_states_from_checkpoint`: False
242
+ - `no_cuda`: False
243
+ - `use_cpu`: False
244
+ - `use_mps_device`: True
245
+ - `seed`: 42
246
+ - `data_seed`: None
247
+ - `jit_mode_eval`: False
248
+ - `bf16`: False
249
+ - `fp16`: False
250
+ - `fp16_opt_level`: O1
251
+ - `half_precision_backend`: auto
252
+ - `bf16_full_eval`: False
253
+ - `fp16_full_eval`: False
254
+ - `tf32`: None
255
+ - `local_rank`: 0
256
+ - `ddp_backend`: None
257
+ - `tpu_num_cores`: None
258
+ - `tpu_metrics_debug`: False
259
+ - `debug`: []
260
+ - `dataloader_drop_last`: False
261
+ - `dataloader_num_workers`: 0
262
+ - `dataloader_prefetch_factor`: None
263
+ - `past_index`: -1
264
+ - `disable_tqdm`: False
265
+ - `remove_unused_columns`: True
266
+ - `label_names`: None
267
+ - `load_best_model_at_end`: False
268
+ - `ignore_data_skip`: False
269
+ - `fsdp`: []
270
+ - `fsdp_min_num_params`: 0
271
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
272
+ - `fsdp_transformer_layer_cls_to_wrap`: None
273
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
274
+ - `parallelism_config`: None
275
+ - `deepspeed`: None
276
+ - `label_smoothing_factor`: 0.0
277
+ - `optim`: adamw_torch_fused
278
+ - `optim_args`: None
279
+ - `adafactor`: False
280
+ - `group_by_length`: False
281
+ - `length_column_name`: length
282
+ - `project`: huggingface
283
+ - `trackio_space_id`: trackio
284
+ - `ddp_find_unused_parameters`: None
285
+ - `ddp_bucket_cap_mb`: None
286
+ - `ddp_broadcast_buffers`: False
287
+ - `dataloader_pin_memory`: False
288
+ - `dataloader_persistent_workers`: False
289
+ - `skip_memory_metrics`: True
290
+ - `use_legacy_prediction_loop`: False
291
+ - `push_to_hub`: False
292
+ - `resume_from_checkpoint`: None
293
+ - `hub_model_id`: None
294
+ - `hub_strategy`: every_save
295
+ - `hub_private_repo`: None
296
+ - `hub_always_push`: False
297
+ - `hub_revision`: None
298
+ - `gradient_checkpointing`: False
299
+ - `gradient_checkpointing_kwargs`: None
300
+ - `include_inputs_for_metrics`: False
301
+ - `include_for_metrics`: []
302
+ - `eval_do_concat_batches`: True
303
+ - `fp16_backend`: auto
304
+ - `push_to_hub_model_id`: None
305
+ - `push_to_hub_organization`: None
306
+ - `mp_parameters`:
307
+ - `auto_find_batch_size`: False
308
+ - `full_determinism`: False
309
+ - `torchdynamo`: None
310
+ - `ray_scope`: last
311
+ - `ddp_timeout`: 1800
312
+ - `torch_compile`: False
313
+ - `torch_compile_backend`: None
314
+ - `torch_compile_mode`: None
315
+ - `include_tokens_per_second`: False
316
+ - `include_num_input_tokens_seen`: no
317
+ - `neftune_noise_alpha`: None
318
+ - `optim_target_modules`: None
319
+ - `batch_eval_metrics`: False
320
+ - `eval_on_start`: False
321
+ - `use_liger_kernel`: False
322
+ - `liger_kernel_config`: None
323
+ - `eval_use_gather_object`: False
324
+ - `average_tokens_across_devices`: True
325
+ - `prompts`: None
326
+ - `batch_sampler`: batch_sampler
327
+ - `multi_dataset_batch_sampler`: proportional
328
+ - `router_mapping`: {}
329
+ - `learning_rate_mapping`: {}
330
+
331
+ </details>
332
+
333
+ ### Training Logs
334
+ | Epoch | Step | Training Loss | Validation Loss |
335
+ |:------:|:----:|:-------------:|:---------------:|
336
+ | 0.1080 | 50 | 1.9486 | - |
337
+ | 0.2160 | 100 | 1.2593 | - |
338
+ | 0.3240 | 150 | 0.7355 | - |
339
+ | 0.4320 | 200 | 0.546 | - |
340
+ | 0.5400 | 250 | 0.4791 | - |
341
+ | 0.6479 | 300 | 0.3971 | - |
342
+ | 0.7559 | 350 | 0.3148 | - |
343
+ | 0.8639 | 400 | 0.3241 | - |
344
+ | 0.9719 | 450 | 0.2552 | - |
345
+ | 1.0 | 463 | - | 0.2700 |
346
+ | 1.0799 | 500 | 0.2515 | - |
347
+ | 1.1879 | 550 | 0.2607 | - |
348
+ | 1.2959 | 600 | 0.2093 | - |
349
+ | 1.4039 | 650 | 0.2755 | - |
350
+ | 1.5119 | 700 | 0.1779 | - |
351
+ | 1.6199 | 750 | 0.1862 | - |
352
+ | 1.7279 | 800 | 0.1645 | - |
353
+ | 1.8359 | 850 | 0.1676 | - |
354
+ | 1.9438 | 900 | 0.2098 | - |
355
+ | 2.0 | 926 | - | 0.1917 |
356
+ | 2.0518 | 950 | 0.2279 | - |
357
+ | 2.1598 | 1000 | 0.1753 | - |
358
+ | 2.2678 | 1050 | 0.181 | - |
359
+ | 2.3758 | 1100 | 0.1755 | - |
360
+ | 2.4838 | 1150 | 0.1633 | - |
361
+ | 2.5918 | 1200 | 0.13 | - |
362
+ | 2.6998 | 1250 | 0.1237 | - |
363
+ | 2.8078 | 1300 | 0.1911 | - |
364
+ | 2.9158 | 1350 | 0.1655 | - |
365
+ | 3.0 | 1389 | - | 0.1829 |
366
+ | 3.0238 | 1400 | 0.1548 | - |
367
+ | 3.1317 | 1450 | 0.1487 | - |
368
+ | 3.2397 | 1500 | 0.1692 | - |
369
+ | 3.3477 | 1550 | 0.1719 | - |
370
+ | 3.4557 | 1600 | 0.1541 | - |
371
+ | 3.5637 | 1650 | 0.1664 | - |
372
+ | 3.6717 | 1700 | 0.1451 | - |
373
+ | 3.7797 | 1750 | 0.163 | - |
374
+ | 3.8877 | 1800 | 0.1615 | - |
375
+ | 3.9957 | 1850 | 0.1445 | - |
376
+ | 4.0 | 1852 | - | 0.1837 |
377
+ | 4.1037 | 1900 | 0.1527 | - |
378
+ | 4.2117 | 1950 | 0.1439 | - |
379
+ | 4.3197 | 2000 | 0.1321 | - |
380
+ | 4.4276 | 2050 | 0.1593 | - |
381
+ | 4.5356 | 2100 | 0.1388 | - |
382
+ | 4.6436 | 2150 | 0.1434 | - |
383
+ | 4.7516 | 2200 | 0.1456 | - |
384
+ | 4.8596 | 2250 | 0.1568 | - |
385
+ | 4.9676 | 2300 | 0.1549 | - |
386
+ | 5.0 | 2315 | - | 0.1620 |
387
+ | 5.0756 | 2350 | 0.1736 | - |
388
+ | 5.1836 | 2400 | 0.1337 | - |
389
+ | 5.2916 | 2450 | 0.1221 | - |
390
+ | 5.3996 | 2500 | 0.1372 | - |
391
+ | 5.5076 | 2550 | 0.1682 | - |
392
+ | 5.6156 | 2600 | 0.1489 | - |
393
+ | 5.7235 | 2650 | 0.138 | - |
394
+ | 5.8315 | 2700 | 0.1197 | - |
395
+ | 5.9395 | 2750 | 0.139 | - |
396
+ | 6.0 | 2778 | - | 0.1673 |
397
+ | 6.0475 | 2800 | 0.1503 | - |
398
+ | 6.1555 | 2850 | 0.1441 | - |
399
+ | 6.2635 | 2900 | 0.1614 | - |
400
+ | 6.3715 | 2950 | 0.1179 | - |
401
+ | 6.4795 | 3000 | 0.1509 | - |
402
+ | 6.5875 | 3050 | 0.1467 | - |
403
+ | 6.6955 | 3100 | 0.141 | - |
404
+ | 6.8035 | 3150 | 0.1358 | - |
405
+ | 6.9114 | 3200 | 0.1639 | - |
406
+ | 7.0 | 3241 | - | 0.1586 |
407
+ | 7.0194 | 3250 | 0.1294 | - |
408
+ | 7.1274 | 3300 | 0.1344 | - |
409
+ | 7.2354 | 3350 | 0.1138 | - |
410
+ | 7.3434 | 3400 | 0.1409 | - |
411
+ | 7.4514 | 3450 | 0.1293 | - |
412
+ | 7.5594 | 3500 | 0.1088 | - |
413
+ | 7.6674 | 3550 | 0.1399 | - |
414
+ | 7.7754 | 3600 | 0.127 | - |
415
+ | 7.8834 | 3650 | 0.1448 | - |
416
+ | 7.9914 | 3700 | 0.1376 | - |
417
+ | 8.0 | 3704 | - | 0.1555 |
418
+ | 8.0994 | 3750 | 0.1171 | - |
419
+ | 8.2073 | 3800 | 0.1221 | - |
420
+ | 8.3153 | 3850 | 0.1406 | - |
421
+ | 8.4233 | 3900 | 0.1299 | - |
422
+ | 8.5313 | 3950 | 0.1363 | - |
423
+ | 8.6393 | 4000 | 0.117 | - |
424
+ | 8.7473 | 4050 | 0.1572 | - |
425
+ | 8.8553 | 4100 | 0.1213 | - |
426
+ | 8.9633 | 4150 | 0.1218 | - |
427
+ | 9.0 | 4167 | - | 0.1551 |
428
+
429
+
430
+ ### Framework Versions
431
+ - Python: 3.9.6
432
+ - Sentence Transformers: 5.1.2
433
+ - Transformers: 4.57.6
434
+ - PyTorch: 2.8.0
435
+ - Accelerate: 1.10.1
436
+ - Datasets: 4.5.0
437
+ - Tokenizers: 0.22.2
438
+
439
+ ## Citation
440
+
441
+ ### BibTeX
442
+
443
+ #### Sentence Transformers
444
+ ```bibtex
445
+ @inproceedings{reimers-2019-sentence-bert,
446
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
447
+ author = "Reimers, Nils and Gurevych, Iryna",
448
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
449
+ month = "11",
450
+ year = "2019",
451
+ publisher = "Association for Computational Linguistics",
452
+ url = "https://arxiv.org/abs/1908.10084",
453
+ }
454
+ ```
455
+
456
+ #### MultipleNegativesRankingLoss
457
+ ```bibtex
458
+ @misc{henderson2017efficient,
459
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
460
+ author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
461
+ year={2017},
462
+ eprint={1705.00652},
463
+ archivePrefix={arXiv},
464
+ primaryClass={cs.CL}
465
+ }
466
+ ```
467
+
468
+ <!--
469
+ ## Glossary
470
+
471
+ *Clearly define terms in order to be accessible across audiences.*
472
+ -->
473
+
474
+ <!--
475
+ ## Model Card Authors
476
+
477
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
478
+ -->
479
+
480
+ <!--
481
+ ## Model Card Contact
482
+
483
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
484
+ -->
checkpoint-4167/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "directionality": "bidi",
8
+ "dtype": "float32",
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-12,
16
+ "max_position_embeddings": 512,
17
+ "model_type": "bert",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 0,
21
+ "pooler_fc_size": 768,
22
+ "pooler_num_attention_heads": 12,
23
+ "pooler_num_fc_layers": 3,
24
+ "pooler_size_per_head": 128,
25
+ "pooler_type": "first_token_transform",
26
+ "position_embedding_type": "absolute",
27
+ "transformers_version": "4.57.6",
28
+ "type_vocab_size": 2,
29
+ "use_cache": true,
30
+ "vocab_size": 21128
31
+ }
checkpoint-4167/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.1.2",
5
+ "transformers": "4.57.6",
6
+ "pytorch": "2.8.0"
7
+ },
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
checkpoint-4167/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43452bb9a584a95ea61f7d4d8470d77953cf349de26458e0d1177b34c4351056
3
+ size 409092920
checkpoint-4167/modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
checkpoint-4167/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:303e5787b5e0edabc7b76d9837fc5f1101c1c6429bd4785d5d92490487352dae
3
+ size 813576203
checkpoint-4167/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c55a2ca55490dabf13b35c6f6d85dd0b3ad463d563509767a49162fe26171ff8
3
+ size 14391
checkpoint-4167/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bca185506ed4a2a196bea5771b05a9d95a11bfe4c4d42ceae6a62e3c9d43dec
3
+ size 1465
checkpoint-4167/sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 128,
3
+ "do_lower_case": false
4
+ }
checkpoint-4167/special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
checkpoint-4167/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-4167/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 128,
51
+ "never_split": null,
52
+ "pad_token": "[PAD]",
53
+ "sep_token": "[SEP]",
54
+ "strip_accents": null,
55
+ "tokenize_chinese_chars": true,
56
+ "tokenizer_class": "BertTokenizer",
57
+ "unk_token": "[UNK]"
58
+ }
checkpoint-4167/trainer_state.json ADDED
@@ -0,0 +1,687 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 9.0,
6
+ "eval_steps": 500,
7
+ "global_step": 4167,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.1079913606911447,
14
+ "grad_norm": 15.24159049987793,
15
+ "learning_rate": 2.1166306695464366e-06,
16
+ "loss": 1.9486,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.2159827213822894,
21
+ "grad_norm": 16.068605422973633,
22
+ "learning_rate": 4.276457883369331e-06,
23
+ "loss": 1.2593,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.32397408207343414,
28
+ "grad_norm": 15.524004936218262,
29
+ "learning_rate": 6.436285097192225e-06,
30
+ "loss": 0.7355,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 0.4319654427645788,
35
+ "grad_norm": 18.918603897094727,
36
+ "learning_rate": 8.59611231101512e-06,
37
+ "loss": 0.546,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 0.5399568034557235,
42
+ "grad_norm": 13.727082252502441,
43
+ "learning_rate": 1.0755939524838012e-05,
44
+ "loss": 0.4791,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 0.6479481641468683,
49
+ "grad_norm": 13.836669921875,
50
+ "learning_rate": 1.2915766738660909e-05,
51
+ "loss": 0.3971,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 0.755939524838013,
56
+ "grad_norm": 4.7911481857299805,
57
+ "learning_rate": 1.5075593952483802e-05,
58
+ "loss": 0.3148,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 0.8639308855291576,
63
+ "grad_norm": 6.807811737060547,
64
+ "learning_rate": 1.7235421166306695e-05,
65
+ "loss": 0.3241,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 0.9719222462203023,
70
+ "grad_norm": 0.5625627040863037,
71
+ "learning_rate": 1.939524838012959e-05,
72
+ "loss": 0.2552,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_loss": 0.2700121998786926,
78
+ "eval_runtime": 20.772,
79
+ "eval_samples_per_second": 39.621,
80
+ "eval_steps_per_second": 2.503,
81
+ "step": 463
82
+ },
83
+ {
84
+ "epoch": 1.079913606911447,
85
+ "grad_norm": 8.922011375427246,
86
+ "learning_rate": 1.982721382289417e-05,
87
+ "loss": 0.2515,
88
+ "step": 500
89
+ },
90
+ {
91
+ "epoch": 1.187904967602592,
92
+ "grad_norm": 3.4080042839050293,
93
+ "learning_rate": 1.9587233021358293e-05,
94
+ "loss": 0.2607,
95
+ "step": 550
96
+ },
97
+ {
98
+ "epoch": 1.2958963282937366,
99
+ "grad_norm": 0.810586154460907,
100
+ "learning_rate": 1.9347252219822416e-05,
101
+ "loss": 0.2093,
102
+ "step": 600
103
+ },
104
+ {
105
+ "epoch": 1.4038876889848813,
106
+ "grad_norm": 4.636417865753174,
107
+ "learning_rate": 1.910727141828654e-05,
108
+ "loss": 0.2755,
109
+ "step": 650
110
+ },
111
+ {
112
+ "epoch": 1.511879049676026,
113
+ "grad_norm": 8.486068725585938,
114
+ "learning_rate": 1.8867290616750663e-05,
115
+ "loss": 0.1779,
116
+ "step": 700
117
+ },
118
+ {
119
+ "epoch": 1.6198704103671706,
120
+ "grad_norm": 13.278620719909668,
121
+ "learning_rate": 1.8627309815214783e-05,
122
+ "loss": 0.1862,
123
+ "step": 750
124
+ },
125
+ {
126
+ "epoch": 1.7278617710583153,
127
+ "grad_norm": 5.251581192016602,
128
+ "learning_rate": 1.8387329013678906e-05,
129
+ "loss": 0.1645,
130
+ "step": 800
131
+ },
132
+ {
133
+ "epoch": 1.83585313174946,
134
+ "grad_norm": 5.251072883605957,
135
+ "learning_rate": 1.814734821214303e-05,
136
+ "loss": 0.1676,
137
+ "step": 850
138
+ },
139
+ {
140
+ "epoch": 1.9438444924406046,
141
+ "grad_norm": 7.933957099914551,
142
+ "learning_rate": 1.7907367410607153e-05,
143
+ "loss": 0.2098,
144
+ "step": 900
145
+ },
146
+ {
147
+ "epoch": 2.0,
148
+ "eval_loss": 0.19168274104595184,
149
+ "eval_runtime": 21.6821,
150
+ "eval_samples_per_second": 37.958,
151
+ "eval_steps_per_second": 2.398,
152
+ "step": 926
153
+ },
154
+ {
155
+ "epoch": 2.0518358531317493,
156
+ "grad_norm": 2.119457960128784,
157
+ "learning_rate": 1.7667386609071277e-05,
158
+ "loss": 0.2279,
159
+ "step": 950
160
+ },
161
+ {
162
+ "epoch": 2.159827213822894,
163
+ "grad_norm": 4.489124774932861,
164
+ "learning_rate": 1.7427405807535397e-05,
165
+ "loss": 0.1753,
166
+ "step": 1000
167
+ },
168
+ {
169
+ "epoch": 2.267818574514039,
170
+ "grad_norm": 5.081973075866699,
171
+ "learning_rate": 1.718742500599952e-05,
172
+ "loss": 0.181,
173
+ "step": 1050
174
+ },
175
+ {
176
+ "epoch": 2.375809935205184,
177
+ "grad_norm": 13.309564590454102,
178
+ "learning_rate": 1.6947444204463643e-05,
179
+ "loss": 0.1755,
180
+ "step": 1100
181
+ },
182
+ {
183
+ "epoch": 2.4838012958963285,
184
+ "grad_norm": 5.2334723472595215,
185
+ "learning_rate": 1.6707463402927767e-05,
186
+ "loss": 0.1633,
187
+ "step": 1150
188
+ },
189
+ {
190
+ "epoch": 2.591792656587473,
191
+ "grad_norm": 4.591087818145752,
192
+ "learning_rate": 1.646748260139189e-05,
193
+ "loss": 0.13,
194
+ "step": 1200
195
+ },
196
+ {
197
+ "epoch": 2.699784017278618,
198
+ "grad_norm": 0.08640618622303009,
199
+ "learning_rate": 1.6227501799856014e-05,
200
+ "loss": 0.1237,
201
+ "step": 1250
202
+ },
203
+ {
204
+ "epoch": 2.8077753779697625,
205
+ "grad_norm": 3.6601691246032715,
206
+ "learning_rate": 1.5987520998320137e-05,
207
+ "loss": 0.1911,
208
+ "step": 1300
209
+ },
210
+ {
211
+ "epoch": 2.915766738660907,
212
+ "grad_norm": 1.4245907068252563,
213
+ "learning_rate": 1.574754019678426e-05,
214
+ "loss": 0.1655,
215
+ "step": 1350
216
+ },
217
+ {
218
+ "epoch": 3.0,
219
+ "eval_loss": 0.1828612983226776,
220
+ "eval_runtime": 22.4072,
221
+ "eval_samples_per_second": 36.729,
222
+ "eval_steps_per_second": 2.321,
223
+ "step": 1389
224
+ },
225
+ {
226
+ "epoch": 3.023758099352052,
227
+ "grad_norm": 3.9116933345794678,
228
+ "learning_rate": 1.550755939524838e-05,
229
+ "loss": 0.1548,
230
+ "step": 1400
231
+ },
232
+ {
233
+ "epoch": 3.1317494600431965,
234
+ "grad_norm": 5.976468563079834,
235
+ "learning_rate": 1.5267578593712504e-05,
236
+ "loss": 0.1487,
237
+ "step": 1450
238
+ },
239
+ {
240
+ "epoch": 3.239740820734341,
241
+ "grad_norm": 3.716071128845215,
242
+ "learning_rate": 1.5027597792176627e-05,
243
+ "loss": 0.1692,
244
+ "step": 1500
245
+ },
246
+ {
247
+ "epoch": 3.347732181425486,
248
+ "grad_norm": 4.578957557678223,
249
+ "learning_rate": 1.478761699064075e-05,
250
+ "loss": 0.1719,
251
+ "step": 1550
252
+ },
253
+ {
254
+ "epoch": 3.4557235421166306,
255
+ "grad_norm": 4.424953460693359,
256
+ "learning_rate": 1.4547636189104872e-05,
257
+ "loss": 0.1541,
258
+ "step": 1600
259
+ },
260
+ {
261
+ "epoch": 3.5637149028077753,
262
+ "grad_norm": 5.738244533538818,
263
+ "learning_rate": 1.4307655387568996e-05,
264
+ "loss": 0.1664,
265
+ "step": 1650
266
+ },
267
+ {
268
+ "epoch": 3.67170626349892,
269
+ "grad_norm": 0.7902525663375854,
270
+ "learning_rate": 1.4067674586033117e-05,
271
+ "loss": 0.1451,
272
+ "step": 1700
273
+ },
274
+ {
275
+ "epoch": 3.7796976241900646,
276
+ "grad_norm": 10.984625816345215,
277
+ "learning_rate": 1.382769378449724e-05,
278
+ "loss": 0.163,
279
+ "step": 1750
280
+ },
281
+ {
282
+ "epoch": 3.8876889848812093,
283
+ "grad_norm": 0.3437069058418274,
284
+ "learning_rate": 1.3587712982961366e-05,
285
+ "loss": 0.1615,
286
+ "step": 1800
287
+ },
288
+ {
289
+ "epoch": 3.995680345572354,
290
+ "grad_norm": 6.087996006011963,
291
+ "learning_rate": 1.3347732181425487e-05,
292
+ "loss": 0.1445,
293
+ "step": 1850
294
+ },
295
+ {
296
+ "epoch": 4.0,
297
+ "eval_loss": 0.18373550474643707,
298
+ "eval_runtime": 19.1948,
299
+ "eval_samples_per_second": 42.876,
300
+ "eval_steps_per_second": 2.709,
301
+ "step": 1852
302
+ },
303
+ {
304
+ "epoch": 4.103671706263499,
305
+ "grad_norm": 9.516853332519531,
306
+ "learning_rate": 1.310775137988961e-05,
307
+ "loss": 0.1527,
308
+ "step": 1900
309
+ },
310
+ {
311
+ "epoch": 4.211663066954643,
312
+ "grad_norm": 3.8790721893310547,
313
+ "learning_rate": 1.2867770578353732e-05,
314
+ "loss": 0.1439,
315
+ "step": 1950
316
+ },
317
+ {
318
+ "epoch": 4.319654427645788,
319
+ "grad_norm": 5.269755840301514,
320
+ "learning_rate": 1.2627789776817856e-05,
321
+ "loss": 0.1321,
322
+ "step": 2000
323
+ },
324
+ {
325
+ "epoch": 4.427645788336933,
326
+ "grad_norm": 6.4369893074035645,
327
+ "learning_rate": 1.238780897528198e-05,
328
+ "loss": 0.1593,
329
+ "step": 2050
330
+ },
331
+ {
332
+ "epoch": 4.535637149028078,
333
+ "grad_norm": 9.754737854003906,
334
+ "learning_rate": 1.2147828173746101e-05,
335
+ "loss": 0.1388,
336
+ "step": 2100
337
+ },
338
+ {
339
+ "epoch": 4.643628509719223,
340
+ "grad_norm": 4.531418800354004,
341
+ "learning_rate": 1.1907847372210224e-05,
342
+ "loss": 0.1434,
343
+ "step": 2150
344
+ },
345
+ {
346
+ "epoch": 4.751619870410368,
347
+ "grad_norm": 2.5565500259399414,
348
+ "learning_rate": 1.1667866570674346e-05,
349
+ "loss": 0.1456,
350
+ "step": 2200
351
+ },
352
+ {
353
+ "epoch": 4.859611231101512,
354
+ "grad_norm": 3.0373988151550293,
355
+ "learning_rate": 1.142788576913847e-05,
356
+ "loss": 0.1568,
357
+ "step": 2250
358
+ },
359
+ {
360
+ "epoch": 4.967602591792657,
361
+ "grad_norm": 7.106463432312012,
362
+ "learning_rate": 1.1187904967602593e-05,
363
+ "loss": 0.1549,
364
+ "step": 2300
365
+ },
366
+ {
367
+ "epoch": 5.0,
368
+ "eval_loss": 0.16200467944145203,
369
+ "eval_runtime": 19.8308,
370
+ "eval_samples_per_second": 41.501,
371
+ "eval_steps_per_second": 2.622,
372
+ "step": 2315
373
+ },
374
+ {
375
+ "epoch": 5.075593952483802,
376
+ "grad_norm": 1.8910789489746094,
377
+ "learning_rate": 1.0947924166066714e-05,
378
+ "loss": 0.1736,
379
+ "step": 2350
380
+ },
381
+ {
382
+ "epoch": 5.183585313174946,
383
+ "grad_norm": 1.3723983764648438,
384
+ "learning_rate": 1.070794336453084e-05,
385
+ "loss": 0.1337,
386
+ "step": 2400
387
+ },
388
+ {
389
+ "epoch": 5.291576673866091,
390
+ "grad_norm": 2.3787972927093506,
391
+ "learning_rate": 1.0467962562994963e-05,
392
+ "loss": 0.1221,
393
+ "step": 2450
394
+ },
395
+ {
396
+ "epoch": 5.399568034557236,
397
+ "grad_norm": 2.458906412124634,
398
+ "learning_rate": 1.0227981761459085e-05,
399
+ "loss": 0.1372,
400
+ "step": 2500
401
+ },
402
+ {
403
+ "epoch": 5.50755939524838,
404
+ "grad_norm": 2.6780011653900146,
405
+ "learning_rate": 9.988000959923208e-06,
406
+ "loss": 0.1682,
407
+ "step": 2550
408
+ },
409
+ {
410
+ "epoch": 5.615550755939525,
411
+ "grad_norm": 2.9658782482147217,
412
+ "learning_rate": 9.74802015838733e-06,
413
+ "loss": 0.1489,
414
+ "step": 2600
415
+ },
416
+ {
417
+ "epoch": 5.72354211663067,
418
+ "grad_norm": 2.0444753170013428,
419
+ "learning_rate": 9.508039356851453e-06,
420
+ "loss": 0.138,
421
+ "step": 2650
422
+ },
423
+ {
424
+ "epoch": 5.831533477321814,
425
+ "grad_norm": 2.001603603363037,
426
+ "learning_rate": 9.268058555315575e-06,
427
+ "loss": 0.1197,
428
+ "step": 2700
429
+ },
430
+ {
431
+ "epoch": 5.939524838012959,
432
+ "grad_norm": 0.0714740976691246,
433
+ "learning_rate": 9.028077753779698e-06,
434
+ "loss": 0.139,
435
+ "step": 2750
436
+ },
437
+ {
438
+ "epoch": 6.0,
439
+ "eval_loss": 0.16732291877269745,
440
+ "eval_runtime": 24.8414,
441
+ "eval_samples_per_second": 33.13,
442
+ "eval_steps_per_second": 2.093,
443
+ "step": 2778
444
+ },
445
+ {
446
+ "epoch": 6.047516198704104,
447
+ "grad_norm": 4.453672885894775,
448
+ "learning_rate": 8.788096952243822e-06,
449
+ "loss": 0.1503,
450
+ "step": 2800
451
+ },
452
+ {
453
+ "epoch": 6.155507559395248,
454
+ "grad_norm": 2.792571544647217,
455
+ "learning_rate": 8.548116150707943e-06,
456
+ "loss": 0.1441,
457
+ "step": 2850
458
+ },
459
+ {
460
+ "epoch": 6.263498920086393,
461
+ "grad_norm": 0.1777549535036087,
462
+ "learning_rate": 8.308135349172067e-06,
463
+ "loss": 0.1614,
464
+ "step": 2900
465
+ },
466
+ {
467
+ "epoch": 6.371490280777538,
468
+ "grad_norm": 3.0961155891418457,
469
+ "learning_rate": 8.06815454763619e-06,
470
+ "loss": 0.1179,
471
+ "step": 2950
472
+ },
473
+ {
474
+ "epoch": 6.479481641468682,
475
+ "grad_norm": 0.10719757527112961,
476
+ "learning_rate": 7.828173746100313e-06,
477
+ "loss": 0.1509,
478
+ "step": 3000
479
+ },
480
+ {
481
+ "epoch": 6.587473002159827,
482
+ "grad_norm": 0.027445131912827492,
483
+ "learning_rate": 7.588192944564435e-06,
484
+ "loss": 0.1467,
485
+ "step": 3050
486
+ },
487
+ {
488
+ "epoch": 6.695464362850972,
489
+ "grad_norm": 5.075443267822266,
490
+ "learning_rate": 7.348212143028558e-06,
491
+ "loss": 0.141,
492
+ "step": 3100
493
+ },
494
+ {
495
+ "epoch": 6.8034557235421165,
496
+ "grad_norm": 0.1375100016593933,
497
+ "learning_rate": 7.108231341492681e-06,
498
+ "loss": 0.1358,
499
+ "step": 3150
500
+ },
501
+ {
502
+ "epoch": 6.911447084233261,
503
+ "grad_norm": 2.0139362812042236,
504
+ "learning_rate": 6.868250539956804e-06,
505
+ "loss": 0.1639,
506
+ "step": 3200
507
+ },
508
+ {
509
+ "epoch": 7.0,
510
+ "eval_loss": 0.15860523283481598,
511
+ "eval_runtime": 26.2207,
512
+ "eval_samples_per_second": 31.387,
513
+ "eval_steps_per_second": 1.983,
514
+ "step": 3241
515
+ },
516
+ {
517
+ "epoch": 7.019438444924406,
518
+ "grad_norm": 0.09312976896762848,
519
+ "learning_rate": 6.628269738420927e-06,
520
+ "loss": 0.1294,
521
+ "step": 3250
522
+ },
523
+ {
524
+ "epoch": 7.1274298056155505,
525
+ "grad_norm": 3.9996461868286133,
526
+ "learning_rate": 6.3882889368850495e-06,
527
+ "loss": 0.1344,
528
+ "step": 3300
529
+ },
530
+ {
531
+ "epoch": 7.235421166306695,
532
+ "grad_norm": 1.266003966331482,
533
+ "learning_rate": 6.148308135349173e-06,
534
+ "loss": 0.1138,
535
+ "step": 3350
536
+ },
537
+ {
538
+ "epoch": 7.34341252699784,
539
+ "grad_norm": 0.8446139097213745,
540
+ "learning_rate": 5.908327333813295e-06,
541
+ "loss": 0.1409,
542
+ "step": 3400
543
+ },
544
+ {
545
+ "epoch": 7.4514038876889845,
546
+ "grad_norm": 2.299039125442505,
547
+ "learning_rate": 5.668346532277418e-06,
548
+ "loss": 0.1293,
549
+ "step": 3450
550
+ },
551
+ {
552
+ "epoch": 7.559395248380129,
553
+ "grad_norm": 1.5416117906570435,
554
+ "learning_rate": 5.428365730741541e-06,
555
+ "loss": 0.1088,
556
+ "step": 3500
557
+ },
558
+ {
559
+ "epoch": 7.667386609071274,
560
+ "grad_norm": 3.3261375427246094,
561
+ "learning_rate": 5.188384929205664e-06,
562
+ "loss": 0.1399,
563
+ "step": 3550
564
+ },
565
+ {
566
+ "epoch": 7.775377969762419,
567
+ "grad_norm": 3.3896985054016113,
568
+ "learning_rate": 4.948404127669787e-06,
569
+ "loss": 0.127,
570
+ "step": 3600
571
+ },
572
+ {
573
+ "epoch": 7.883369330453563,
574
+ "grad_norm": 0.08087552338838577,
575
+ "learning_rate": 4.70842332613391e-06,
576
+ "loss": 0.1448,
577
+ "step": 3650
578
+ },
579
+ {
580
+ "epoch": 7.991360691144708,
581
+ "grad_norm": 0.712371289730072,
582
+ "learning_rate": 4.468442524598032e-06,
583
+ "loss": 0.1376,
584
+ "step": 3700
585
+ },
586
+ {
587
+ "epoch": 8.0,
588
+ "eval_loss": 0.1555214673280716,
589
+ "eval_runtime": 26.2282,
590
+ "eval_samples_per_second": 31.378,
591
+ "eval_steps_per_second": 1.983,
592
+ "step": 3704
593
+ },
594
+ {
595
+ "epoch": 8.099352051835853,
596
+ "grad_norm": 0.04458538815379143,
597
+ "learning_rate": 4.228461723062156e-06,
598
+ "loss": 0.1171,
599
+ "step": 3750
600
+ },
601
+ {
602
+ "epoch": 8.207343412526997,
603
+ "grad_norm": 3.261307954788208,
604
+ "learning_rate": 3.988480921526278e-06,
605
+ "loss": 0.1221,
606
+ "step": 3800
607
+ },
608
+ {
609
+ "epoch": 8.315334773218142,
610
+ "grad_norm": 0.07829932123422623,
611
+ "learning_rate": 3.7485001199904008e-06,
612
+ "loss": 0.1406,
613
+ "step": 3850
614
+ },
615
+ {
616
+ "epoch": 8.423326133909287,
617
+ "grad_norm": 2.330026626586914,
618
+ "learning_rate": 3.508519318454524e-06,
619
+ "loss": 0.1299,
620
+ "step": 3900
621
+ },
622
+ {
623
+ "epoch": 8.531317494600431,
624
+ "grad_norm": 0.7942795753479004,
625
+ "learning_rate": 3.2685385169186467e-06,
626
+ "loss": 0.1363,
627
+ "step": 3950
628
+ },
629
+ {
630
+ "epoch": 8.639308855291576,
631
+ "grad_norm": 2.6522200107574463,
632
+ "learning_rate": 3.0285577153827692e-06,
633
+ "loss": 0.117,
634
+ "step": 4000
635
+ },
636
+ {
637
+ "epoch": 8.74730021598272,
638
+ "grad_norm": 2.3869872093200684,
639
+ "learning_rate": 2.7885769138468926e-06,
640
+ "loss": 0.1572,
641
+ "step": 4050
642
+ },
643
+ {
644
+ "epoch": 8.855291576673865,
645
+ "grad_norm": 1.1808924674987793,
646
+ "learning_rate": 2.548596112311015e-06,
647
+ "loss": 0.1213,
648
+ "step": 4100
649
+ },
650
+ {
651
+ "epoch": 8.96328293736501,
652
+ "grad_norm": 3.1210787296295166,
653
+ "learning_rate": 2.308615310775138e-06,
654
+ "loss": 0.1218,
655
+ "step": 4150
656
+ },
657
+ {
658
+ "epoch": 9.0,
659
+ "eval_loss": 0.15509691834449768,
660
+ "eval_runtime": 26.4137,
661
+ "eval_samples_per_second": 31.158,
662
+ "eval_steps_per_second": 1.969,
663
+ "step": 4167
664
+ }
665
+ ],
666
+ "logging_steps": 50,
667
+ "max_steps": 4630,
668
+ "num_input_tokens_seen": 0,
669
+ "num_train_epochs": 10,
670
+ "save_steps": 500,
671
+ "stateful_callbacks": {
672
+ "TrainerControl": {
673
+ "args": {
674
+ "should_epoch_stop": false,
675
+ "should_evaluate": false,
676
+ "should_log": false,
677
+ "should_save": true,
678
+ "should_training_stop": false
679
+ },
680
+ "attributes": {}
681
+ }
682
+ },
683
+ "total_flos": 0.0,
684
+ "train_batch_size": 16,
685
+ "trial_name": null,
686
+ "trial_params": null
687
+ }
checkpoint-4167/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89b5fdf7428645df3050766edd4c89caa34ed38abdf6736d5be07796db4cff8e
3
+ size 6225
checkpoint-4167/vocab.txt ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-4630/1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
checkpoint-4630/README.md ADDED
@@ -0,0 +1,497 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:7400
9
+ - loss:MultipleNegativesRankingLoss
10
+ base_model: shibing624/text2vec-base-chinese
11
+ widget:
12
+ - source_sentence: 在一条无限长的均匀管道中,流体在任何截面的行为都一样。这种沿管道方向的不变性直接保证了流量在管道中处处相等。正是因为没有哪个位置特殊,流体才不会在某处堆积或消散。
13
+ sentences:
14
+ - 一条法律规则如果对所有人一视同仁——不因身份、地位而区别对待——那么在这条规则下,所有人的权利总量保持均衡。平等对待的对称性直接导致权利分配的守恒。
15
+ - 一个极少地震的地区,地震仪几乎永远显示平静,每天的读数不传递新信息。但在地震活跃带,每次仪器读数都可能是不同强度和方向的震动,每条数据都在揭示地壳应力的新变化,监测价值远高于前者。
16
+ - 下棋时,你每一步都假设对手会走对他最有利的着法,然后在这种假设下选择对自己最有利的走法。对手也这样想。双方都做最坏打算后的最优选择,最终棋局会走向一个双方都无法改善的局面。
17
+ - source_sentence: 一条河流遭受污染后,下游的污染物浓度逐渐降低。最初几公里下降很快,但之后稀释速度放缓,残余污染在很长的河段上维持着低但可检测的浓度,远比预期消散得更慢。
18
+ sentences:
19
+ - 新修一条地铁线路后,沿线房价和人口不会立刻变化。三五年后居民逐渐迁入,商业配套慢慢成熟,片区终于繁荣起来。但此时交通压力又超出设计承载力,需要再次扩建,而扩建的效果又要等很多年才显现。
20
+ - 知识从教师集中的城市向师资匮乏的乡村扩散——通过支教、在线课程、教材传播。知识资源丰富的地方像高浓度溶液,自然向匮乏之处渗透,差距越大渗透的动力就越强。
21
+ - 一只股票在利空消息后暴跌,第一天跌幅最大。之后几天继续下跌但幅度递减,随后进入漫长的阴跌期——每天只跌一点点,但持续很久。这种缓慢出血比暴跌更让投资者煎熬。
22
+ - source_sentence: 今天的气温很大程度上取决于昨天和前天的气温,再加上一些随机扰动——比如一阵突然的冷风。天气预报之所以短期比较准、长期越来越不靠谱,就是因为每天的天气都是前几天天气的延续加上不可预测的新干扰的叠加。
23
+ sentences:
24
+ - 所有博主都在晚上八点发视频,因为这是流量高峰。如果你换到其他时段,观众更少;如果大家都在八点发,竞争虽然激烈但流量也最大。每个博主都没有动力单独改变发布时间。
25
+ - 脑电图记录了头皮表面的微弱电信号。把这个复杂信号拆解成频率成分后,发现不同状态下主导频率不同:清醒专注时以中高频为主,放松时以中低频为主,深睡时以极低频为主。大脑的活动状态可以通过频率成分的变化来判读。
26
+ - 今年的粮食产量与前几年有关——连续丰年后土壤肥力可能下降,连续歉收后农民会调整种植策略。每年的天气条件是不可预测的随机因素。农业产量的年际变化就是历史趋势的延续叠加上气候随机性的结果。
27
+ - source_sentence: 某些金属合金在高温下柔软易弯。当温度降到某个值时,内部结构突然重新排列,金属瞬间变得坚硬而脆。升温时恢复柔软的温度却比冷却时变硬的温度高出不少,两个转变温度之间存在明显的不对称。
28
+ sentences:
29
+ - 有机化学家要从简单原料合成目标药物分子,每一步反应都有不同的产率和成本。在所有可能的合成路线中,他们要找到步骤最少、总产率最高的那条反应路径。
30
+ - 热钱在全球市场中有一个主流方向——从低利率国家流向高利率国家。但每一笔具体的资金流向又带有投资者个体判断的随机性。宏观的利差驱动和微观的个体分散叠加,形成了资本流动的全貌。
31
+ - 暖湿气流和冷空气对峙时,天气在阴沉和放晴之间维持微妙的平衡。当冷空气势力刚好越过某个强度,锋面突然南压,天气从温暖直接跳变为寒潮。锋面过后气温回升的过程却是缓慢渐进的,冷暖转换不对等。
32
+ - source_sentence: 肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。
33
+ sentences:
34
+ - 在一个人人收入相同的社会,你知道一个人的收入不会获得任何关于他身份的信息。但在收入差距大且分布均匀的社会,每了解一个人的收入都能帮你推断他的职业、教育和生活方式,信息量丰富得多。
35
+ - 国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。
36
+ - 金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。
37
+ pipeline_tag: sentence-similarity
38
+ library_name: sentence-transformers
39
+ ---
40
+
41
+ # SentenceTransformer based on shibing624/text2vec-base-chinese
42
+
43
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese). It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
44
+
45
+ ## Model Details
46
+
47
+ ### Model Description
48
+ - **Model Type:** Sentence Transformer
49
+ - **Base model:** [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese) <!-- at revision 183bb99aa7af74355fb58d16edf8c13ae7c5433e -->
50
+ - **Maximum Sequence Length:** 128 tokens
51
+ - **Output Dimensionality:** 768 dimensions
52
+ - **Similarity Function:** Cosine Similarity
53
+ <!-- - **Training Dataset:** Unknown -->
54
+ <!-- - **Language:** Unknown -->
55
+ <!-- - **License:** Unknown -->
56
+
57
+ ### Model Sources
58
+
59
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
60
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
61
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
62
+
63
+ ### Full Model Architecture
64
+
65
+ ```
66
+ SentenceTransformer(
67
+ (0): Transformer({'max_seq_length': 128, 'do_lower_case': False, 'architecture': 'BertModel'})
68
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
69
+ )
70
+ ```
71
+
72
+ ## Usage
73
+
74
+ ### Direct Usage (Sentence Transformers)
75
+
76
+ First install the Sentence Transformers library:
77
+
78
+ ```bash
79
+ pip install -U sentence-transformers
80
+ ```
81
+
82
+ Then you can load this model and run inference.
83
+ ```python
84
+ from sentence_transformers import SentenceTransformer
85
+
86
+ # Download from the 🤗 Hub
87
+ model = SentenceTransformer("sentence_transformers_model_id")
88
+ # Run inference
89
+ sentences = [
90
+ '肥皂泡总是圆形的,因为球体是在给定体积下表面积最小的形状。肥皂膜的表面张力驱动它自动找到使总表面能量最低的形态。无需谁来计算,物理规律自动将泡泡推向这个最优形状。',
91
+ '金属冷却凝固时晶粒的边界不是随机的。晶粒界面自动调整到使总界面能量最小的构型。三个晶粒相遇的角度总是接近一百二十度,这不是巧合而是能量最优化的必然结果。',
92
+ '国际冲突的爆发呈现出明显的丛集效应:和平年代可能持续几十年,然后短期内多场战争密集爆发。第一次世界大战结束二十年后就爆发了第二次。冲突不是随机均匀分布在历史时间轴上的,而是倾向于成簇出现。',
93
+ ]
94
+ embeddings = model.encode(sentences)
95
+ print(embeddings.shape)
96
+ # [3, 768]
97
+
98
+ # Get the similarity scores for the embeddings
99
+ similarities = model.similarity(embeddings, embeddings)
100
+ print(similarities)
101
+ # tensor([[1.0000, 0.9134, 0.0324],
102
+ # [0.9134, 1.0000, 0.1733],
103
+ # [0.0324, 0.1733, 1.0000]])
104
+ ```
105
+
106
+ <!--
107
+ ### Direct Usage (Transformers)
108
+
109
+ <details><summary>Click to see the direct usage in Transformers</summary>
110
+
111
+ </details>
112
+ -->
113
+
114
+ <!--
115
+ ### Downstream Usage (Sentence Transformers)
116
+
117
+ You can finetune this model on your own dataset.
118
+
119
+ <details><summary>Click to expand</summary>
120
+
121
+ </details>
122
+ -->
123
+
124
+ <!--
125
+ ### Out-of-Scope Use
126
+
127
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
128
+ -->
129
+
130
+ <!--
131
+ ## Bias, Risks and Limitations
132
+
133
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
134
+ -->
135
+
136
+ <!--
137
+ ### Recommendations
138
+
139
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
140
+ -->
141
+
142
+ ## Training Details
143
+
144
+ ### Training Dataset
145
+
146
+ #### Unnamed Dataset
147
+
148
+ * Size: 7,400 training samples
149
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
150
+ * Approximate statistics based on the first 1000 samples:
151
+ | | sentence1 | sentence2 | label |
152
+ |:--------|:-----------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
153
+ | type | string | string | float |
154
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.8 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.35 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
155
+ * Samples:
156
+ | sentence1 | sentence2 | label |
157
+ |:-----------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------|:-----------------|
158
+ | <code>急诊室里病人到达的时间没有任何规律。可能一个小时来六个人,下一个小时只来一个。但每个人来不来和其他人完全无关,平均下来每小时到达人数基本稳定。等待时间有时很短有时很长。</code> | <code>保险公司收到的理赔申请在时间上零散分布。本周三个下周可能七个,具体哪天收到完全不可预测。但年度总理赔数很稳定,每一起事故独立发生,一起理赔不影响下一起何时出现。</code> | <code>1.0</code> |
159
+ | <code>经济系统中几十个指标——GDP、失业率、通胀率——互相影响看似复杂。但分析后发现,它们的联动可以归结为几个独立的方向:一个方向代表经济周期,另一个方向代表结构调整。沿着这些方向看,经济的行为最简单。</code> | <code>股市里几千只股票每天的涨跌看似无序,但分析后发现大部分波动可以归结为几个主要方向——一个是大盘整体涨跌,一个是行业板块轮动。沿着这几个方向看,市场的行为变得最简单最可预测。</code> | <code>1.0</code> |
160
+ | <code>天气中既有持续几小时的雷暴这种短时高频事件,也有持续几天的冷锋这种中期过程,还有季节变化这种长期低频趋势。气象分析需要在不同时间尺度上同时观察——用小窗口看雷暴的精确时间,用大窗口看季节的整体趋势。</code> | <code>说话时辅音是短促的高频爆发持续几十毫秒,元音是持续较长的低频振动。语音识别需要同时精确定位短促辅音的出现时刻和长元音的频率特征。用短窗口分析辅音的时间细节,用长窗口分析元音的频率细节。</code> | <code>1.0</code> |
161
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
162
+ ```json
163
+ {
164
+ "scale": 20.0,
165
+ "similarity_fct": "cos_sim",
166
+ "gather_across_devices": false
167
+ }
168
+ ```
169
+
170
+ ### Evaluation Dataset
171
+
172
+ #### Unnamed Dataset
173
+
174
+ * Size: 823 evaluation samples
175
+ * Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
176
+ * Approximate statistics based on the first 823 samples:
177
+ | | sentence1 | sentence2 | label |
178
+ |:--------|:------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:--------------------------------------------------------------|
179
+ | type | string | string | float |
180
+ | details | <ul><li>min: 66 tokens</li><li>mean: 88.98 tokens</li><li>max: 115 tokens</li></ul> | <ul><li>min: 66 tokens</li><li>mean: 88.22 tokens</li><li>max: 121 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
181
+ * Samples:
182
+ | sentence1 | sentence2 | label |
183
+ |:----------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------|:-----------------|
184
+ | <code>一个人今天的情绪状态很大程度上延续昨天的情绪。心情好的惯性会持续几天,心情差也会低落一阵。但每天都可能遇到随机事件——收到好消息或坏消息——把情绪轨迹推离原来的方向。情绪的变化就是历史惯性加上每天的随机冲击。</code> | <code>一段文章中下一个词的选择很大程度上取决于前面几个词。说了因为就很可能接所以,说了虽然就很可能接但是。但作者偶尔会突破预期用一个意外的词制造效果。文本的生成像是对已有上下文的延续加上创造性的随机变化。</code> | <code>1.0</code> |
185
+ | <code>一个信号可以用随时间变化的波形来描述,也可以用各频率成分的强度来描述。这两种描述是同一信号的镜像刻画。在时间域里难以解决的问题转到频率域可能很简单,反之亦然。两种视角提供了看同一问题的对称方式。</code> | <code>国际贸易谈判中,出口国追求的高价恰好是进口国要避免的高价。出口方的最优策略和进口方的最优策略是同一个博弈的镜像刻画。谈判桌两边的分析虽然立场相反,但使用的逻辑框架完全对称。</code> | <code>1.0</code> |
186
+ | <code>一个社区从高度凝聚走向分散化的过程中,并不是某天突然崩解。在转折期,各种规模的社交圈子此起彼伏地形成又解散,小到三五个人大到上百人的聚集同时存在,群体行为的波动达到最大,各种规模的组织形态都能观察到。</code> | <code>某种合金在特定温度附近,微观结构表现出异常行为。原子排列的有序和无序区域在各个尺度上共存,从纳米到微米都能看到两种状态的混合。材料的比热容在这个温度附近出现异常尖峰,而且不同成分的合金在各自的临界温度附近表现出惊人的相似规律。</code> | <code>1.0</code> |
187
+ * Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
188
+ ```json
189
+ {
190
+ "scale": 20.0,
191
+ "similarity_fct": "cos_sim",
192
+ "gather_across_devices": false
193
+ }
194
+ ```
195
+
196
+ ### Training Hyperparameters
197
+ #### Non-Default Hyperparameters
198
+
199
+ - `eval_strategy`: epoch
200
+ - `per_device_train_batch_size`: 16
201
+ - `per_device_eval_batch_size`: 16
202
+ - `learning_rate`: 2e-05
203
+ - `num_train_epochs`: 10
204
+ - `warmup_ratio`: 0.1
205
+ - `use_mps_device`: True
206
+ - `dataloader_pin_memory`: False
207
+
208
+ #### All Hyperparameters
209
+ <details><summary>Click to expand</summary>
210
+
211
+ - `overwrite_output_dir`: False
212
+ - `do_predict`: False
213
+ - `eval_strategy`: epoch
214
+ - `prediction_loss_only`: True
215
+ - `per_device_train_batch_size`: 16
216
+ - `per_device_eval_batch_size`: 16
217
+ - `per_gpu_train_batch_size`: None
218
+ - `per_gpu_eval_batch_size`: None
219
+ - `gradient_accumulation_steps`: 1
220
+ - `eval_accumulation_steps`: None
221
+ - `torch_empty_cache_steps`: None
222
+ - `learning_rate`: 2e-05
223
+ - `weight_decay`: 0.0
224
+ - `adam_beta1`: 0.9
225
+ - `adam_beta2`: 0.999
226
+ - `adam_epsilon`: 1e-08
227
+ - `max_grad_norm`: 1.0
228
+ - `num_train_epochs`: 10
229
+ - `max_steps`: -1
230
+ - `lr_scheduler_type`: linear
231
+ - `lr_scheduler_kwargs`: None
232
+ - `warmup_ratio`: 0.1
233
+ - `warmup_steps`: 0
234
+ - `log_level`: passive
235
+ - `log_level_replica`: warning
236
+ - `log_on_each_node`: True
237
+ - `logging_nan_inf_filter`: True
238
+ - `save_safetensors`: True
239
+ - `save_on_each_node`: False
240
+ - `save_only_model`: False
241
+ - `restore_callback_states_from_checkpoint`: False
242
+ - `no_cuda`: False
243
+ - `use_cpu`: False
244
+ - `use_mps_device`: True
245
+ - `seed`: 42
246
+ - `data_seed`: None
247
+ - `jit_mode_eval`: False
248
+ - `bf16`: False
249
+ - `fp16`: False
250
+ - `fp16_opt_level`: O1
251
+ - `half_precision_backend`: auto
252
+ - `bf16_full_eval`: False
253
+ - `fp16_full_eval`: False
254
+ - `tf32`: None
255
+ - `local_rank`: 0
256
+ - `ddp_backend`: None
257
+ - `tpu_num_cores`: None
258
+ - `tpu_metrics_debug`: False
259
+ - `debug`: []
260
+ - `dataloader_drop_last`: False
261
+ - `dataloader_num_workers`: 0
262
+ - `dataloader_prefetch_factor`: None
263
+ - `past_index`: -1
264
+ - `disable_tqdm`: False
265
+ - `remove_unused_columns`: True
266
+ - `label_names`: None
267
+ - `load_best_model_at_end`: False
268
+ - `ignore_data_skip`: False
269
+ - `fsdp`: []
270
+ - `fsdp_min_num_params`: 0
271
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
272
+ - `fsdp_transformer_layer_cls_to_wrap`: None
273
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
274
+ - `parallelism_config`: None
275
+ - `deepspeed`: None
276
+ - `label_smoothing_factor`: 0.0
277
+ - `optim`: adamw_torch_fused
278
+ - `optim_args`: None
279
+ - `adafactor`: False
280
+ - `group_by_length`: False
281
+ - `length_column_name`: length
282
+ - `project`: huggingface
283
+ - `trackio_space_id`: trackio
284
+ - `ddp_find_unused_parameters`: None
285
+ - `ddp_bucket_cap_mb`: None
286
+ - `ddp_broadcast_buffers`: False
287
+ - `dataloader_pin_memory`: False
288
+ - `dataloader_persistent_workers`: False
289
+ - `skip_memory_metrics`: True
290
+ - `use_legacy_prediction_loop`: False
291
+ - `push_to_hub`: False
292
+ - `resume_from_checkpoint`: None
293
+ - `hub_model_id`: None
294
+ - `hub_strategy`: every_save
295
+ - `hub_private_repo`: None
296
+ - `hub_always_push`: False
297
+ - `hub_revision`: None
298
+ - `gradient_checkpointing`: False
299
+ - `gradient_checkpointing_kwargs`: None
300
+ - `include_inputs_for_metrics`: False
301
+ - `include_for_metrics`: []
302
+ - `eval_do_concat_batches`: True
303
+ - `fp16_backend`: auto
304
+ - `push_to_hub_model_id`: None
305
+ - `push_to_hub_organization`: None
306
+ - `mp_parameters`:
307
+ - `auto_find_batch_size`: False
308
+ - `full_determinism`: False
309
+ - `torchdynamo`: None
310
+ - `ray_scope`: last
311
+ - `ddp_timeout`: 1800
312
+ - `torch_compile`: False
313
+ - `torch_compile_backend`: None
314
+ - `torch_compile_mode`: None
315
+ - `include_tokens_per_second`: False
316
+ - `include_num_input_tokens_seen`: no
317
+ - `neftune_noise_alpha`: None
318
+ - `optim_target_modules`: None
319
+ - `batch_eval_metrics`: False
320
+ - `eval_on_start`: False
321
+ - `use_liger_kernel`: False
322
+ - `liger_kernel_config`: None
323
+ - `eval_use_gather_object`: False
324
+ - `average_tokens_across_devices`: True
325
+ - `prompts`: None
326
+ - `batch_sampler`: batch_sampler
327
+ - `multi_dataset_batch_sampler`: proportional
328
+ - `router_mapping`: {}
329
+ - `learning_rate_mapping`: {}
330
+
331
+ </details>
332
+
333
+ ### Training Logs
334
+ <details><summary>Click to expand</summary>
335
+
336
+ | Epoch | Step | Training Loss | Validation Loss |
337
+ |:------:|:----:|:-------------:|:---------------:|
338
+ | 0.1080 | 50 | 1.9486 | - |
339
+ | 0.2160 | 100 | 1.2593 | - |
340
+ | 0.3240 | 150 | 0.7355 | - |
341
+ | 0.4320 | 200 | 0.546 | - |
342
+ | 0.5400 | 250 | 0.4791 | - |
343
+ | 0.6479 | 300 | 0.3971 | - |
344
+ | 0.7559 | 350 | 0.3148 | - |
345
+ | 0.8639 | 400 | 0.3241 | - |
346
+ | 0.9719 | 450 | 0.2552 | - |
347
+ | 1.0 | 463 | - | 0.2700 |
348
+ | 1.0799 | 500 | 0.2515 | - |
349
+ | 1.1879 | 550 | 0.2607 | - |
350
+ | 1.2959 | 600 | 0.2093 | - |
351
+ | 1.4039 | 650 | 0.2755 | - |
352
+ | 1.5119 | 700 | 0.1779 | - |
353
+ | 1.6199 | 750 | 0.1862 | - |
354
+ | 1.7279 | 800 | 0.1645 | - |
355
+ | 1.8359 | 850 | 0.1676 | - |
356
+ | 1.9438 | 900 | 0.2098 | - |
357
+ | 2.0 | 926 | - | 0.1917 |
358
+ | 2.0518 | 950 | 0.2279 | - |
359
+ | 2.1598 | 1000 | 0.1753 | - |
360
+ | 2.2678 | 1050 | 0.181 | - |
361
+ | 2.3758 | 1100 | 0.1755 | - |
362
+ | 2.4838 | 1150 | 0.1633 | - |
363
+ | 2.5918 | 1200 | 0.13 | - |
364
+ | 2.6998 | 1250 | 0.1237 | - |
365
+ | 2.8078 | 1300 | 0.1911 | - |
366
+ | 2.9158 | 1350 | 0.1655 | - |
367
+ | 3.0 | 1389 | - | 0.1829 |
368
+ | 3.0238 | 1400 | 0.1548 | - |
369
+ | 3.1317 | 1450 | 0.1487 | - |
370
+ | 3.2397 | 1500 | 0.1692 | - |
371
+ | 3.3477 | 1550 | 0.1719 | - |
372
+ | 3.4557 | 1600 | 0.1541 | - |
373
+ | 3.5637 | 1650 | 0.1664 | - |
374
+ | 3.6717 | 1700 | 0.1451 | - |
375
+ | 3.7797 | 1750 | 0.163 | - |
376
+ | 3.8877 | 1800 | 0.1615 | - |
377
+ | 3.9957 | 1850 | 0.1445 | - |
378
+ | 4.0 | 1852 | - | 0.1837 |
379
+ | 4.1037 | 1900 | 0.1527 | - |
380
+ | 4.2117 | 1950 | 0.1439 | - |
381
+ | 4.3197 | 2000 | 0.1321 | - |
382
+ | 4.4276 | 2050 | 0.1593 | - |
383
+ | 4.5356 | 2100 | 0.1388 | - |
384
+ | 4.6436 | 2150 | 0.1434 | - |
385
+ | 4.7516 | 2200 | 0.1456 | - |
386
+ | 4.8596 | 2250 | 0.1568 | - |
387
+ | 4.9676 | 2300 | 0.1549 | - |
388
+ | 5.0 | 2315 | - | 0.1620 |
389
+ | 5.0756 | 2350 | 0.1736 | - |
390
+ | 5.1836 | 2400 | 0.1337 | - |
391
+ | 5.2916 | 2450 | 0.1221 | - |
392
+ | 5.3996 | 2500 | 0.1372 | - |
393
+ | 5.5076 | 2550 | 0.1682 | - |
394
+ | 5.6156 | 2600 | 0.1489 | - |
395
+ | 5.7235 | 2650 | 0.138 | - |
396
+ | 5.8315 | 2700 | 0.1197 | - |
397
+ | 5.9395 | 2750 | 0.139 | - |
398
+ | 6.0 | 2778 | - | 0.1673 |
399
+ | 6.0475 | 2800 | 0.1503 | - |
400
+ | 6.1555 | 2850 | 0.1441 | - |
401
+ | 6.2635 | 2900 | 0.1614 | - |
402
+ | 6.3715 | 2950 | 0.1179 | - |
403
+ | 6.4795 | 3000 | 0.1509 | - |
404
+ | 6.5875 | 3050 | 0.1467 | - |
405
+ | 6.6955 | 3100 | 0.141 | - |
406
+ | 6.8035 | 3150 | 0.1358 | - |
407
+ | 6.9114 | 3200 | 0.1639 | - |
408
+ | 7.0 | 3241 | - | 0.1586 |
409
+ | 7.0194 | 3250 | 0.1294 | - |
410
+ | 7.1274 | 3300 | 0.1344 | - |
411
+ | 7.2354 | 3350 | 0.1138 | - |
412
+ | 7.3434 | 3400 | 0.1409 | - |
413
+ | 7.4514 | 3450 | 0.1293 | - |
414
+ | 7.5594 | 3500 | 0.1088 | - |
415
+ | 7.6674 | 3550 | 0.1399 | - |
416
+ | 7.7754 | 3600 | 0.127 | - |
417
+ | 7.8834 | 3650 | 0.1448 | - |
418
+ | 7.9914 | 3700 | 0.1376 | - |
419
+ | 8.0 | 3704 | - | 0.1555 |
420
+ | 8.0994 | 3750 | 0.1171 | - |
421
+ | 8.2073 | 3800 | 0.1221 | - |
422
+ | 8.3153 | 3850 | 0.1406 | - |
423
+ | 8.4233 | 3900 | 0.1299 | - |
424
+ | 8.5313 | 3950 | 0.1363 | - |
425
+ | 8.6393 | 4000 | 0.117 | - |
426
+ | 8.7473 | 4050 | 0.1572 | - |
427
+ | 8.8553 | 4100 | 0.1213 | - |
428
+ | 8.9633 | 4150 | 0.1218 | - |
429
+ | 9.0 | 4167 | - | 0.1551 |
430
+ | 9.0713 | 4200 | 0.1333 | - |
431
+ | 9.1793 | 4250 | 0.1339 | - |
432
+ | 9.2873 | 4300 | 0.1477 | - |
433
+ | 9.3952 | 4350 | 0.1013 | - |
434
+ | 9.5032 | 4400 | 0.1241 | - |
435
+ | 9.6112 | 4450 | 0.1091 | - |
436
+ | 9.7192 | 4500 | 0.1077 | - |
437
+ | 9.8272 | 4550 | 0.1261 | - |
438
+ | 9.9352 | 4600 | 0.1396 | - |
439
+ | 10.0 | 4630 | - | 0.1545 |
440
+
441
+ </details>
442
+
443
+ ### Framework Versions
444
+ - Python: 3.9.6
445
+ - Sentence Transformers: 5.1.2
446
+ - Transformers: 4.57.6
447
+ - PyTorch: 2.8.0
448
+ - Accelerate: 1.10.1
449
+ - Datasets: 4.5.0
450
+ - Tokenizers: 0.22.2
451
+
452
+ ## Citation
453
+
454
+ ### BibTeX
455
+
456
+ #### Sentence Transformers
457
+ ```bibtex
458
+ @inproceedings{reimers-2019-sentence-bert,
459
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
460
+ author = "Reimers, Nils and Gurevych, Iryna",
461
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
462
+ month = "11",
463
+ year = "2019",
464
+ publisher = "Association for Computational Linguistics",
465
+ url = "https://arxiv.org/abs/1908.10084",
466
+ }
467
+ ```
468
+
469
+ #### MultipleNegativesRankingLoss
470
+ ```bibtex
471
+ @misc{henderson2017efficient,
472
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
473
+ author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
474
+ year={2017},
475
+ eprint={1705.00652},
476
+ archivePrefix={arXiv},
477
+ primaryClass={cs.CL}
478
+ }
479
+ ```
480
+
481
+ <!--
482
+ ## Glossary
483
+
484
+ *Clearly define terms in order to be accessible across audiences.*
485
+ -->
486
+
487
+ <!--
488
+ ## Model Card Authors
489
+
490
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
491
+ -->
492
+
493
+ <!--
494
+ ## Model Card Contact
495
+
496
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
497
+ -->
checkpoint-4630/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "directionality": "bidi",
8
+ "dtype": "float32",
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-12,
16
+ "max_position_embeddings": 512,
17
+ "model_type": "bert",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 0,
21
+ "pooler_fc_size": 768,
22
+ "pooler_num_attention_heads": 12,
23
+ "pooler_num_fc_layers": 3,
24
+ "pooler_size_per_head": 128,
25
+ "pooler_type": "first_token_transform",
26
+ "position_embedding_type": "absolute",
27
+ "transformers_version": "4.57.6",
28
+ "type_vocab_size": 2,
29
+ "use_cache": true,
30
+ "vocab_size": 21128
31
+ }
checkpoint-4630/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.1.2",
5
+ "transformers": "4.57.6",
6
+ "pytorch": "2.8.0"
7
+ },
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
checkpoint-4630/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:176d19577b1586e060b24aefa7b6c74cadd066fee235432470757688cc6aab57
3
+ size 409092920
checkpoint-4630/modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
checkpoint-4630/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d4159b411c0340279c887fe8f54973303ee818e80523fa6630995675afad12a
3
+ size 813576203
checkpoint-4630/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:418af22fe0bdf619942a292bd60916334d8fa9fbcc21d27e00e65fc2f7c18b08
3
+ size 14391
checkpoint-4630/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7046ed943782e16609102b299ea824f4a0be5cf7b2e570f381ff563c94662953
3
+ size 1465
checkpoint-4630/sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 128,
3
+ "do_lower_case": false
4
+ }
checkpoint-4630/special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
checkpoint-4630/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-4630/tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 128,
51
+ "never_split": null,
52
+ "pad_token": "[PAD]",
53
+ "sep_token": "[SEP]",
54
+ "strip_accents": null,
55
+ "tokenize_chinese_chars": true,
56
+ "tokenizer_class": "BertTokenizer",
57
+ "unk_token": "[UNK]"
58
+ }
checkpoint-4630/trainer_state.json ADDED
@@ -0,0 +1,758 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 10.0,
6
+ "eval_steps": 500,
7
+ "global_step": 4630,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.1079913606911447,
14
+ "grad_norm": 15.24159049987793,
15
+ "learning_rate": 2.1166306695464366e-06,
16
+ "loss": 1.9486,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.2159827213822894,
21
+ "grad_norm": 16.068605422973633,
22
+ "learning_rate": 4.276457883369331e-06,
23
+ "loss": 1.2593,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.32397408207343414,
28
+ "grad_norm": 15.524004936218262,
29
+ "learning_rate": 6.436285097192225e-06,
30
+ "loss": 0.7355,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 0.4319654427645788,
35
+ "grad_norm": 18.918603897094727,
36
+ "learning_rate": 8.59611231101512e-06,
37
+ "loss": 0.546,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 0.5399568034557235,
42
+ "grad_norm": 13.727082252502441,
43
+ "learning_rate": 1.0755939524838012e-05,
44
+ "loss": 0.4791,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 0.6479481641468683,
49
+ "grad_norm": 13.836669921875,
50
+ "learning_rate": 1.2915766738660909e-05,
51
+ "loss": 0.3971,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 0.755939524838013,
56
+ "grad_norm": 4.7911481857299805,
57
+ "learning_rate": 1.5075593952483802e-05,
58
+ "loss": 0.3148,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 0.8639308855291576,
63
+ "grad_norm": 6.807811737060547,
64
+ "learning_rate": 1.7235421166306695e-05,
65
+ "loss": 0.3241,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 0.9719222462203023,
70
+ "grad_norm": 0.5625627040863037,
71
+ "learning_rate": 1.939524838012959e-05,
72
+ "loss": 0.2552,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_loss": 0.2700121998786926,
78
+ "eval_runtime": 20.772,
79
+ "eval_samples_per_second": 39.621,
80
+ "eval_steps_per_second": 2.503,
81
+ "step": 463
82
+ },
83
+ {
84
+ "epoch": 1.079913606911447,
85
+ "grad_norm": 8.922011375427246,
86
+ "learning_rate": 1.982721382289417e-05,
87
+ "loss": 0.2515,
88
+ "step": 500
89
+ },
90
+ {
91
+ "epoch": 1.187904967602592,
92
+ "grad_norm": 3.4080042839050293,
93
+ "learning_rate": 1.9587233021358293e-05,
94
+ "loss": 0.2607,
95
+ "step": 550
96
+ },
97
+ {
98
+ "epoch": 1.2958963282937366,
99
+ "grad_norm": 0.810586154460907,
100
+ "learning_rate": 1.9347252219822416e-05,
101
+ "loss": 0.2093,
102
+ "step": 600
103
+ },
104
+ {
105
+ "epoch": 1.4038876889848813,
106
+ "grad_norm": 4.636417865753174,
107
+ "learning_rate": 1.910727141828654e-05,
108
+ "loss": 0.2755,
109
+ "step": 650
110
+ },
111
+ {
112
+ "epoch": 1.511879049676026,
113
+ "grad_norm": 8.486068725585938,
114
+ "learning_rate": 1.8867290616750663e-05,
115
+ "loss": 0.1779,
116
+ "step": 700
117
+ },
118
+ {
119
+ "epoch": 1.6198704103671706,
120
+ "grad_norm": 13.278620719909668,
121
+ "learning_rate": 1.8627309815214783e-05,
122
+ "loss": 0.1862,
123
+ "step": 750
124
+ },
125
+ {
126
+ "epoch": 1.7278617710583153,
127
+ "grad_norm": 5.251581192016602,
128
+ "learning_rate": 1.8387329013678906e-05,
129
+ "loss": 0.1645,
130
+ "step": 800
131
+ },
132
+ {
133
+ "epoch": 1.83585313174946,
134
+ "grad_norm": 5.251072883605957,
135
+ "learning_rate": 1.814734821214303e-05,
136
+ "loss": 0.1676,
137
+ "step": 850
138
+ },
139
+ {
140
+ "epoch": 1.9438444924406046,
141
+ "grad_norm": 7.933957099914551,
142
+ "learning_rate": 1.7907367410607153e-05,
143
+ "loss": 0.2098,
144
+ "step": 900
145
+ },
146
+ {
147
+ "epoch": 2.0,
148
+ "eval_loss": 0.19168274104595184,
149
+ "eval_runtime": 21.6821,
150
+ "eval_samples_per_second": 37.958,
151
+ "eval_steps_per_second": 2.398,
152
+ "step": 926
153
+ },
154
+ {
155
+ "epoch": 2.0518358531317493,
156
+ "grad_norm": 2.119457960128784,
157
+ "learning_rate": 1.7667386609071277e-05,
158
+ "loss": 0.2279,
159
+ "step": 950
160
+ },
161
+ {
162
+ "epoch": 2.159827213822894,
163
+ "grad_norm": 4.489124774932861,
164
+ "learning_rate": 1.7427405807535397e-05,
165
+ "loss": 0.1753,
166
+ "step": 1000
167
+ },
168
+ {
169
+ "epoch": 2.267818574514039,
170
+ "grad_norm": 5.081973075866699,
171
+ "learning_rate": 1.718742500599952e-05,
172
+ "loss": 0.181,
173
+ "step": 1050
174
+ },
175
+ {
176
+ "epoch": 2.375809935205184,
177
+ "grad_norm": 13.309564590454102,
178
+ "learning_rate": 1.6947444204463643e-05,
179
+ "loss": 0.1755,
180
+ "step": 1100
181
+ },
182
+ {
183
+ "epoch": 2.4838012958963285,
184
+ "grad_norm": 5.2334723472595215,
185
+ "learning_rate": 1.6707463402927767e-05,
186
+ "loss": 0.1633,
187
+ "step": 1150
188
+ },
189
+ {
190
+ "epoch": 2.591792656587473,
191
+ "grad_norm": 4.591087818145752,
192
+ "learning_rate": 1.646748260139189e-05,
193
+ "loss": 0.13,
194
+ "step": 1200
195
+ },
196
+ {
197
+ "epoch": 2.699784017278618,
198
+ "grad_norm": 0.08640618622303009,
199
+ "learning_rate": 1.6227501799856014e-05,
200
+ "loss": 0.1237,
201
+ "step": 1250
202
+ },
203
+ {
204
+ "epoch": 2.8077753779697625,
205
+ "grad_norm": 3.6601691246032715,
206
+ "learning_rate": 1.5987520998320137e-05,
207
+ "loss": 0.1911,
208
+ "step": 1300
209
+ },
210
+ {
211
+ "epoch": 2.915766738660907,
212
+ "grad_norm": 1.4245907068252563,
213
+ "learning_rate": 1.574754019678426e-05,
214
+ "loss": 0.1655,
215
+ "step": 1350
216
+ },
217
+ {
218
+ "epoch": 3.0,
219
+ "eval_loss": 0.1828612983226776,
220
+ "eval_runtime": 22.4072,
221
+ "eval_samples_per_second": 36.729,
222
+ "eval_steps_per_second": 2.321,
223
+ "step": 1389
224
+ },
225
+ {
226
+ "epoch": 3.023758099352052,
227
+ "grad_norm": 3.9116933345794678,
228
+ "learning_rate": 1.550755939524838e-05,
229
+ "loss": 0.1548,
230
+ "step": 1400
231
+ },
232
+ {
233
+ "epoch": 3.1317494600431965,
234
+ "grad_norm": 5.976468563079834,
235
+ "learning_rate": 1.5267578593712504e-05,
236
+ "loss": 0.1487,
237
+ "step": 1450
238
+ },
239
+ {
240
+ "epoch": 3.239740820734341,
241
+ "grad_norm": 3.716071128845215,
242
+ "learning_rate": 1.5027597792176627e-05,
243
+ "loss": 0.1692,
244
+ "step": 1500
245
+ },
246
+ {
247
+ "epoch": 3.347732181425486,
248
+ "grad_norm": 4.578957557678223,
249
+ "learning_rate": 1.478761699064075e-05,
250
+ "loss": 0.1719,
251
+ "step": 1550
252
+ },
253
+ {
254
+ "epoch": 3.4557235421166306,
255
+ "grad_norm": 4.424953460693359,
256
+ "learning_rate": 1.4547636189104872e-05,
257
+ "loss": 0.1541,
258
+ "step": 1600
259
+ },
260
+ {
261
+ "epoch": 3.5637149028077753,
262
+ "grad_norm": 5.738244533538818,
263
+ "learning_rate": 1.4307655387568996e-05,
264
+ "loss": 0.1664,
265
+ "step": 1650
266
+ },
267
+ {
268
+ "epoch": 3.67170626349892,
269
+ "grad_norm": 0.7902525663375854,
270
+ "learning_rate": 1.4067674586033117e-05,
271
+ "loss": 0.1451,
272
+ "step": 1700
273
+ },
274
+ {
275
+ "epoch": 3.7796976241900646,
276
+ "grad_norm": 10.984625816345215,
277
+ "learning_rate": 1.382769378449724e-05,
278
+ "loss": 0.163,
279
+ "step": 1750
280
+ },
281
+ {
282
+ "epoch": 3.8876889848812093,
283
+ "grad_norm": 0.3437069058418274,
284
+ "learning_rate": 1.3587712982961366e-05,
285
+ "loss": 0.1615,
286
+ "step": 1800
287
+ },
288
+ {
289
+ "epoch": 3.995680345572354,
290
+ "grad_norm": 6.087996006011963,
291
+ "learning_rate": 1.3347732181425487e-05,
292
+ "loss": 0.1445,
293
+ "step": 1850
294
+ },
295
+ {
296
+ "epoch": 4.0,
297
+ "eval_loss": 0.18373550474643707,
298
+ "eval_runtime": 19.1948,
299
+ "eval_samples_per_second": 42.876,
300
+ "eval_steps_per_second": 2.709,
301
+ "step": 1852
302
+ },
303
+ {
304
+ "epoch": 4.103671706263499,
305
+ "grad_norm": 9.516853332519531,
306
+ "learning_rate": 1.310775137988961e-05,
307
+ "loss": 0.1527,
308
+ "step": 1900
309
+ },
310
+ {
311
+ "epoch": 4.211663066954643,
312
+ "grad_norm": 3.8790721893310547,
313
+ "learning_rate": 1.2867770578353732e-05,
314
+ "loss": 0.1439,
315
+ "step": 1950
316
+ },
317
+ {
318
+ "epoch": 4.319654427645788,
319
+ "grad_norm": 5.269755840301514,
320
+ "learning_rate": 1.2627789776817856e-05,
321
+ "loss": 0.1321,
322
+ "step": 2000
323
+ },
324
+ {
325
+ "epoch": 4.427645788336933,
326
+ "grad_norm": 6.4369893074035645,
327
+ "learning_rate": 1.238780897528198e-05,
328
+ "loss": 0.1593,
329
+ "step": 2050
330
+ },
331
+ {
332
+ "epoch": 4.535637149028078,
333
+ "grad_norm": 9.754737854003906,
334
+ "learning_rate": 1.2147828173746101e-05,
335
+ "loss": 0.1388,
336
+ "step": 2100
337
+ },
338
+ {
339
+ "epoch": 4.643628509719223,
340
+ "grad_norm": 4.531418800354004,
341
+ "learning_rate": 1.1907847372210224e-05,
342
+ "loss": 0.1434,
343
+ "step": 2150
344
+ },
345
+ {
346
+ "epoch": 4.751619870410368,
347
+ "grad_norm": 2.5565500259399414,
348
+ "learning_rate": 1.1667866570674346e-05,
349
+ "loss": 0.1456,
350
+ "step": 2200
351
+ },
352
+ {
353
+ "epoch": 4.859611231101512,
354
+ "grad_norm": 3.0373988151550293,
355
+ "learning_rate": 1.142788576913847e-05,
356
+ "loss": 0.1568,
357
+ "step": 2250
358
+ },
359
+ {
360
+ "epoch": 4.967602591792657,
361
+ "grad_norm": 7.106463432312012,
362
+ "learning_rate": 1.1187904967602593e-05,
363
+ "loss": 0.1549,
364
+ "step": 2300
365
+ },
366
+ {
367
+ "epoch": 5.0,
368
+ "eval_loss": 0.16200467944145203,
369
+ "eval_runtime": 19.8308,
370
+ "eval_samples_per_second": 41.501,
371
+ "eval_steps_per_second": 2.622,
372
+ "step": 2315
373
+ },
374
+ {
375
+ "epoch": 5.075593952483802,
376
+ "grad_norm": 1.8910789489746094,
377
+ "learning_rate": 1.0947924166066714e-05,
378
+ "loss": 0.1736,
379
+ "step": 2350
380
+ },
381
+ {
382
+ "epoch": 5.183585313174946,
383
+ "grad_norm": 1.3723983764648438,
384
+ "learning_rate": 1.070794336453084e-05,
385
+ "loss": 0.1337,
386
+ "step": 2400
387
+ },
388
+ {
389
+ "epoch": 5.291576673866091,
390
+ "grad_norm": 2.3787972927093506,
391
+ "learning_rate": 1.0467962562994963e-05,
392
+ "loss": 0.1221,
393
+ "step": 2450
394
+ },
395
+ {
396
+ "epoch": 5.399568034557236,
397
+ "grad_norm": 2.458906412124634,
398
+ "learning_rate": 1.0227981761459085e-05,
399
+ "loss": 0.1372,
400
+ "step": 2500
401
+ },
402
+ {
403
+ "epoch": 5.50755939524838,
404
+ "grad_norm": 2.6780011653900146,
405
+ "learning_rate": 9.988000959923208e-06,
406
+ "loss": 0.1682,
407
+ "step": 2550
408
+ },
409
+ {
410
+ "epoch": 5.615550755939525,
411
+ "grad_norm": 2.9658782482147217,
412
+ "learning_rate": 9.74802015838733e-06,
413
+ "loss": 0.1489,
414
+ "step": 2600
415
+ },
416
+ {
417
+ "epoch": 5.72354211663067,
418
+ "grad_norm": 2.0444753170013428,
419
+ "learning_rate": 9.508039356851453e-06,
420
+ "loss": 0.138,
421
+ "step": 2650
422
+ },
423
+ {
424
+ "epoch": 5.831533477321814,
425
+ "grad_norm": 2.001603603363037,
426
+ "learning_rate": 9.268058555315575e-06,
427
+ "loss": 0.1197,
428
+ "step": 2700
429
+ },
430
+ {
431
+ "epoch": 5.939524838012959,
432
+ "grad_norm": 0.0714740976691246,
433
+ "learning_rate": 9.028077753779698e-06,
434
+ "loss": 0.139,
435
+ "step": 2750
436
+ },
437
+ {
438
+ "epoch": 6.0,
439
+ "eval_loss": 0.16732291877269745,
440
+ "eval_runtime": 24.8414,
441
+ "eval_samples_per_second": 33.13,
442
+ "eval_steps_per_second": 2.093,
443
+ "step": 2778
444
+ },
445
+ {
446
+ "epoch": 6.047516198704104,
447
+ "grad_norm": 4.453672885894775,
448
+ "learning_rate": 8.788096952243822e-06,
449
+ "loss": 0.1503,
450
+ "step": 2800
451
+ },
452
+ {
453
+ "epoch": 6.155507559395248,
454
+ "grad_norm": 2.792571544647217,
455
+ "learning_rate": 8.548116150707943e-06,
456
+ "loss": 0.1441,
457
+ "step": 2850
458
+ },
459
+ {
460
+ "epoch": 6.263498920086393,
461
+ "grad_norm": 0.1777549535036087,
462
+ "learning_rate": 8.308135349172067e-06,
463
+ "loss": 0.1614,
464
+ "step": 2900
465
+ },
466
+ {
467
+ "epoch": 6.371490280777538,
468
+ "grad_norm": 3.0961155891418457,
469
+ "learning_rate": 8.06815454763619e-06,
470
+ "loss": 0.1179,
471
+ "step": 2950
472
+ },
473
+ {
474
+ "epoch": 6.479481641468682,
475
+ "grad_norm": 0.10719757527112961,
476
+ "learning_rate": 7.828173746100313e-06,
477
+ "loss": 0.1509,
478
+ "step": 3000
479
+ },
480
+ {
481
+ "epoch": 6.587473002159827,
482
+ "grad_norm": 0.027445131912827492,
483
+ "learning_rate": 7.588192944564435e-06,
484
+ "loss": 0.1467,
485
+ "step": 3050
486
+ },
487
+ {
488
+ "epoch": 6.695464362850972,
489
+ "grad_norm": 5.075443267822266,
490
+ "learning_rate": 7.348212143028558e-06,
491
+ "loss": 0.141,
492
+ "step": 3100
493
+ },
494
+ {
495
+ "epoch": 6.8034557235421165,
496
+ "grad_norm": 0.1375100016593933,
497
+ "learning_rate": 7.108231341492681e-06,
498
+ "loss": 0.1358,
499
+ "step": 3150
500
+ },
501
+ {
502
+ "epoch": 6.911447084233261,
503
+ "grad_norm": 2.0139362812042236,
504
+ "learning_rate": 6.868250539956804e-06,
505
+ "loss": 0.1639,
506
+ "step": 3200
507
+ },
508
+ {
509
+ "epoch": 7.0,
510
+ "eval_loss": 0.15860523283481598,
511
+ "eval_runtime": 26.2207,
512
+ "eval_samples_per_second": 31.387,
513
+ "eval_steps_per_second": 1.983,
514
+ "step": 3241
515
+ },
516
+ {
517
+ "epoch": 7.019438444924406,
518
+ "grad_norm": 0.09312976896762848,
519
+ "learning_rate": 6.628269738420927e-06,
520
+ "loss": 0.1294,
521
+ "step": 3250
522
+ },
523
+ {
524
+ "epoch": 7.1274298056155505,
525
+ "grad_norm": 3.9996461868286133,
526
+ "learning_rate": 6.3882889368850495e-06,
527
+ "loss": 0.1344,
528
+ "step": 3300
529
+ },
530
+ {
531
+ "epoch": 7.235421166306695,
532
+ "grad_norm": 1.266003966331482,
533
+ "learning_rate": 6.148308135349173e-06,
534
+ "loss": 0.1138,
535
+ "step": 3350
536
+ },
537
+ {
538
+ "epoch": 7.34341252699784,
539
+ "grad_norm": 0.8446139097213745,
540
+ "learning_rate": 5.908327333813295e-06,
541
+ "loss": 0.1409,
542
+ "step": 3400
543
+ },
544
+ {
545
+ "epoch": 7.4514038876889845,
546
+ "grad_norm": 2.299039125442505,
547
+ "learning_rate": 5.668346532277418e-06,
548
+ "loss": 0.1293,
549
+ "step": 3450
550
+ },
551
+ {
552
+ "epoch": 7.559395248380129,
553
+ "grad_norm": 1.5416117906570435,
554
+ "learning_rate": 5.428365730741541e-06,
555
+ "loss": 0.1088,
556
+ "step": 3500
557
+ },
558
+ {
559
+ "epoch": 7.667386609071274,
560
+ "grad_norm": 3.3261375427246094,
561
+ "learning_rate": 5.188384929205664e-06,
562
+ "loss": 0.1399,
563
+ "step": 3550
564
+ },
565
+ {
566
+ "epoch": 7.775377969762419,
567
+ "grad_norm": 3.3896985054016113,
568
+ "learning_rate": 4.948404127669787e-06,
569
+ "loss": 0.127,
570
+ "step": 3600
571
+ },
572
+ {
573
+ "epoch": 7.883369330453563,
574
+ "grad_norm": 0.08087552338838577,
575
+ "learning_rate": 4.70842332613391e-06,
576
+ "loss": 0.1448,
577
+ "step": 3650
578
+ },
579
+ {
580
+ "epoch": 7.991360691144708,
581
+ "grad_norm": 0.712371289730072,
582
+ "learning_rate": 4.468442524598032e-06,
583
+ "loss": 0.1376,
584
+ "step": 3700
585
+ },
586
+ {
587
+ "epoch": 8.0,
588
+ "eval_loss": 0.1555214673280716,
589
+ "eval_runtime": 26.2282,
590
+ "eval_samples_per_second": 31.378,
591
+ "eval_steps_per_second": 1.983,
592
+ "step": 3704
593
+ },
594
+ {
595
+ "epoch": 8.099352051835853,
596
+ "grad_norm": 0.04458538815379143,
597
+ "learning_rate": 4.228461723062156e-06,
598
+ "loss": 0.1171,
599
+ "step": 3750
600
+ },
601
+ {
602
+ "epoch": 8.207343412526997,
603
+ "grad_norm": 3.261307954788208,
604
+ "learning_rate": 3.988480921526278e-06,
605
+ "loss": 0.1221,
606
+ "step": 3800
607
+ },
608
+ {
609
+ "epoch": 8.315334773218142,
610
+ "grad_norm": 0.07829932123422623,
611
+ "learning_rate": 3.7485001199904008e-06,
612
+ "loss": 0.1406,
613
+ "step": 3850
614
+ },
615
+ {
616
+ "epoch": 8.423326133909287,
617
+ "grad_norm": 2.330026626586914,
618
+ "learning_rate": 3.508519318454524e-06,
619
+ "loss": 0.1299,
620
+ "step": 3900
621
+ },
622
+ {
623
+ "epoch": 8.531317494600431,
624
+ "grad_norm": 0.7942795753479004,
625
+ "learning_rate": 3.2685385169186467e-06,
626
+ "loss": 0.1363,
627
+ "step": 3950
628
+ },
629
+ {
630
+ "epoch": 8.639308855291576,
631
+ "grad_norm": 2.6522200107574463,
632
+ "learning_rate": 3.0285577153827692e-06,
633
+ "loss": 0.117,
634
+ "step": 4000
635
+ },
636
+ {
637
+ "epoch": 8.74730021598272,
638
+ "grad_norm": 2.3869872093200684,
639
+ "learning_rate": 2.7885769138468926e-06,
640
+ "loss": 0.1572,
641
+ "step": 4050
642
+ },
643
+ {
644
+ "epoch": 8.855291576673865,
645
+ "grad_norm": 1.1808924674987793,
646
+ "learning_rate": 2.548596112311015e-06,
647
+ "loss": 0.1213,
648
+ "step": 4100
649
+ },
650
+ {
651
+ "epoch": 8.96328293736501,
652
+ "grad_norm": 3.1210787296295166,
653
+ "learning_rate": 2.308615310775138e-06,
654
+ "loss": 0.1218,
655
+ "step": 4150
656
+ },
657
+ {
658
+ "epoch": 9.0,
659
+ "eval_loss": 0.15509691834449768,
660
+ "eval_runtime": 26.4137,
661
+ "eval_samples_per_second": 31.158,
662
+ "eval_steps_per_second": 1.969,
663
+ "step": 4167
664
+ },
665
+ {
666
+ "epoch": 9.071274298056155,
667
+ "grad_norm": 1.0357710123062134,
668
+ "learning_rate": 2.068634509239261e-06,
669
+ "loss": 0.1333,
670
+ "step": 4200
671
+ },
672
+ {
673
+ "epoch": 9.1792656587473,
674
+ "grad_norm": 3.8207991123199463,
675
+ "learning_rate": 1.8286537077033838e-06,
676
+ "loss": 0.1339,
677
+ "step": 4250
678
+ },
679
+ {
680
+ "epoch": 9.287257019438446,
681
+ "grad_norm": 3.1769232749938965,
682
+ "learning_rate": 1.5886729061675068e-06,
683
+ "loss": 0.1477,
684
+ "step": 4300
685
+ },
686
+ {
687
+ "epoch": 9.39524838012959,
688
+ "grad_norm": 2.3620474338531494,
689
+ "learning_rate": 1.3486921046316296e-06,
690
+ "loss": 0.1013,
691
+ "step": 4350
692
+ },
693
+ {
694
+ "epoch": 9.503239740820735,
695
+ "grad_norm": 0.047901928424835205,
696
+ "learning_rate": 1.1087113030957525e-06,
697
+ "loss": 0.1241,
698
+ "step": 4400
699
+ },
700
+ {
701
+ "epoch": 9.61123110151188,
702
+ "grad_norm": 0.07165756076574326,
703
+ "learning_rate": 8.687305015598753e-07,
704
+ "loss": 0.1091,
705
+ "step": 4450
706
+ },
707
+ {
708
+ "epoch": 9.719222462203025,
709
+ "grad_norm": 2.574002265930176,
710
+ "learning_rate": 6.287497000239981e-07,
711
+ "loss": 0.1077,
712
+ "step": 4500
713
+ },
714
+ {
715
+ "epoch": 9.82721382289417,
716
+ "grad_norm": 1.7075910568237305,
717
+ "learning_rate": 3.88768898488121e-07,
718
+ "loss": 0.1261,
719
+ "step": 4550
720
+ },
721
+ {
722
+ "epoch": 9.935205183585314,
723
+ "grad_norm": 2.4924235343933105,
724
+ "learning_rate": 1.4878809695224384e-07,
725
+ "loss": 0.1396,
726
+ "step": 4600
727
+ },
728
+ {
729
+ "epoch": 10.0,
730
+ "eval_loss": 0.1544954627752304,
731
+ "eval_runtime": 27.5689,
732
+ "eval_samples_per_second": 29.852,
733
+ "eval_steps_per_second": 1.886,
734
+ "step": 4630
735
+ }
736
+ ],
737
+ "logging_steps": 50,
738
+ "max_steps": 4630,
739
+ "num_input_tokens_seen": 0,
740
+ "num_train_epochs": 10,
741
+ "save_steps": 500,
742
+ "stateful_callbacks": {
743
+ "TrainerControl": {
744
+ "args": {
745
+ "should_epoch_stop": false,
746
+ "should_evaluate": false,
747
+ "should_log": false,
748
+ "should_save": true,
749
+ "should_training_stop": true
750
+ },
751
+ "attributes": {}
752
+ }
753
+ },
754
+ "total_flos": 0.0,
755
+ "train_batch_size": 16,
756
+ "trial_name": null,
757
+ "trial_params": null
758
+ }
checkpoint-4630/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89b5fdf7428645df3050766edd4c89caa34ed38abdf6736d5be07796db4cff8e
3
+ size 6225
checkpoint-4630/vocab.txt ADDED
The diff for this file is too large to render. See raw diff
 
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "directionality": "bidi",
8
+ "dtype": "float32",
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-12,
16
+ "max_position_embeddings": 512,
17
+ "model_type": "bert",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 0,
21
+ "pooler_fc_size": 768,
22
+ "pooler_num_attention_heads": 12,
23
+ "pooler_num_fc_layers": 3,
24
+ "pooler_size_per_head": 128,
25
+ "pooler_type": "first_token_transform",
26
+ "position_embedding_type": "absolute",
27
+ "transformers_version": "4.57.6",
28
+ "type_vocab_size": 2,
29
+ "use_cache": true,
30
+ "vocab_size": 21128
31
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.1.2",
5
+ "transformers": "4.57.6",
6
+ "pytorch": "2.8.0"
7
+ },
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:176d19577b1586e060b24aefa7b6c74cadd066fee235432470757688cc6aab57
3
+ size 409092920
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 128,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 128,
51
+ "never_split": null,
52
+ "pad_token": "[PAD]",
53
+ "sep_token": "[SEP]",
54
+ "strip_accents": null,
55
+ "tokenize_chinese_chars": true,
56
+ "tokenizer_class": "BertTokenizer",
57
+ "unk_token": "[UNK]"
58
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff