George2002 commited on
Commit
efcf241
·
verified ·
1 Parent(s): acf3a40

Upload model checkpoint

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 1024,
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,639 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - generated_from_trainer
7
+ - dataset_size:41115
8
+ - loss:TripletLoss
9
+ base_model: intfloat/multilingual-e5-large
10
+ widget:
11
+ - source_sentence: 'query: Какое заверение требуется для разрешения Финансового управляющего
12
+ на снятие всех ограничений?'
13
+ sentences:
14
+ - 'passage: Выберите причину розыска?
15
+
16
+
17
+ Интересует компенсация по вкладам 1991 года
18
+
19
+
20
+ Сотрудник проверяет у клиента наличие оригинала сберкнижки или иного документа
21
+ (копия книжки/дубликат книжки/договор вклада/квитанция ф. 31/банковский ордер
22
+ или банковская справка, выписка/вкладчик получал компенсацию ранее по данному
23
+ вкладу) (далее - Сберкнижка).'
24
+ - "passage: При обращении клиента - банкрота для выполнения платежей по Плану реструктуризации\
25
+ \ долгов, он должен дополнительно предъявить: \n\n- Решение суда об утверждении\
26
+ \ Плана реструктуризации долгов;\n- Разрешение ФУ на снятие блокировок со счетов,\
27
+ \ отличных от Специального банковского счета (в случае необходимости)\n- Разрешение\
28
+ \ ФУ на снятие всех ограничений и возможность пользоваться всеми банковскими продуктами\
29
+ \ \n\n\n!!!!! Разрешение Финансового управляющего должно быть заверено личной\
30
+ \ подписью ФУ, копия удостоверена нотариусом (печать ФУ при наличии).\n\n\n\n\n\
31
+ Предоставлено разрешение ФУ на снятие всех ограничений \n\nДля снятия всех ограничений\
32
+ \ и для проведения платежей по \"Плану реструктуризации долгов\" необходимо\n\
33
+ Завести заявку через кнопку \"Решить вопрос клиента\" \n\nПо тематике: Банкротство\
34
+ \ ФЛ\n\nПрикрепить файлы сканов: \n - Решение АС о признании обоснованным заявления\
35
+ \ о признании гражданина банкротом и введения реструктуризации его долгов, \n\
36
+ \ + Решение АС об утверждении финансового управляющего (это может быть один документ),\
37
+ \ \n - Решение АС об утверждении Плана РД\n - Письменное разрешение/согласие финансового\
38
+ \ управляющего на снятие всех огранничений с продуктов должника с целью исполнения\
39
+ \ Плана РД. \n\n \nПосле снятия арестов, должник сам распоряжается счетами и\
40
+ \ выполняет операции согласно Плана реструктуризации долгов."
41
+ - "passage: Для снятия всех ограничений необходимо предоставить Решение АС о завершении/прекращении\
42
+ \ банкротства.\n\n\n* В решении арбитражного суда могут быть указаны следующие\
43
+ \ определения:\n\n- Завершена реализация имущества;\n- Прекращено банкротство\
44
+ \ (в том числе заключение мирового соглашения);\n- Завершено конкурсное производство.\n\
45
+ \nКлиент обратился впервые\n\nСотрудник запрашивает у клиента Решение суда о завершении/\
46
+ \ прекращении процедуры банкротства (Копию). \n\nЕсли по клиенту закончилась процедура\
47
+ \ реструктуризация долгов и клиент пользовался банковской картой в качестве специального\
48
+ \ банковского счета с ежемесячным лимитом 50 000.00, необходимо проверить наличие\
49
+ \ установленного по карте лимита в АС ФС, и при его наличии, убрать выставленный\
50
+ \ лимит."
51
+ - source_sentence: 'query: Каковы шаги для оформления заявки на разблокировку счета
52
+ в АС ''Сбердруг''?'
53
+ sentences:
54
+ - "passage: Проверить наличие ареста на счет, с которого необходимо произвести выдачу\n\
55
+ \n\nЕСТЬ АРЕСТ на счете\n\nОформить заявку в АС \"Сбердруг\" на разблокировку\
56
+ \ счета\n1.\tОткрыть Каталог –> \nОбслуживание клиентов –> \nОперационный центр\
57
+ \ –> \nСопровождение операций Физических лиц –>\nФЛ. Работа с операциями Банкротов,\
58
+ \ откроется шаблон заявки.\n2.\tЗаполнить поля шаблона заявки в соответствии:\
59
+ \ \nНомер ТБ: Номер ТБ, ведения счета \nВыберите тематику запроса: Взаимодействие\
60
+ \ с ВСП/ОСБ\nТип запроса: Разблокировка счетов\nТема запроса: Разблокировка счетов\
61
+ \ для выдачи наличных\nДалее указать:\nФИО банкрота - ФИО банкрота \nНомер счета\
62
+ \ - Номер счета/вклада \nОбратился - (выбор Банкрот/ ФУ)\nВид получения\
63
+ \ (выбор нарочно / почта)\nВнутренний клиент (указать фамилию сотрудника,\
64
+ \ который сможет отследить исполнение заявки в случае отсутствия сотрудника, зарегистрировавшего\
65
+ \ заявку) \nСообщение в случае необходимости\n\n\nПрикрепить файлы сканов: ДУЛ\
66
+ \ клиента+ иные документы\nи нажать зеленую кнопку \"ОТПРАВИТЬ\".\nЗапрос направлен\
67
+ \ на исполнение.\n\nОригиналы всех предоставленных и оформленных в процессе подготовки\
68
+ \ заявки документов остаются у клиента."
69
+ - 'passage: Кто обратился?
70
+
71
+
72
+ Вкладчик/Представитель вкладчика (полномочия подтверждены)
73
+
74
+
75
+ Выберите причину розыска?'
76
+ - "passage: Сотрудник формирует заявку в \"Сбердруг\" по шаблону:\nКаталог→\nОбслуживание\
77
+ \ клиентов –> \nОперационный центр –> \nСопровождение операций Физических лиц\
78
+ \ –>\nФЛ. Работа с операциями Банкротов, откроется шаблон заявки\n\nЗаполнить\
79
+ \ поля шаблона заявки в соответствии: \nНомер ТБ: Номер ТБ, ведения счета (Для\
80
+ \ более быстрой обработки заявки верно указывайте ТБ клиента)\nВыберите тематику\
81
+ \ запроса: Взаимодействие с ВСП/ОСБ\n\nТип запроса: Разблокировка счетов\nТема\
82
+ \ запроса: Завершение процедуры банкротства\n\nДалее указать:\nФИО банкрота -\
83
+ \ ФИО банкрота \nНомер счета - Номер счета/вклада \nОбратился - (выбор\
84
+ \ Банкрот/ ФУ) \nВид получения - (выбор нарочно / почта)\nСообщение\
85
+ \ заполнить в случае необходимости указания особенностей\n\nК заявке приложить\
86
+ \ скан-копии:\n - Определение арбитражного суда о прекращении/завершении банкротства\n\
87
+ \ - ДУЛ Клиента\nнажать зеленую кнопку \"ОТПРАВИТЬ\".\nЗапрос направлен на исполнение.\n\
88
+ \n\nПредупредить клиента, что мероприятия по снятию ограничений с продуктов будут\
89
+ \ выполнены в течении 7 рабочих дней.\nОригиналы предоставленных документов остаются\
90
+ \ у клиента.\n\n!!!!!ОБЯЗАТЕЛЬНО ВЫБРАТЬ ВЕРНУЮ ТЕМУ ЗАПРОСА!!!!! Тема запроса:\
91
+ \ Завершение процедуры банкротства\nЕсли выбрать тему запроса: разблокировка счета,\
92
+ \ счет может заблокировать повторно \n\n\nВ случае ��сли блокирован СБОЛ клиента\
93
+ \ дополнительно\nСотрудник регистрирует обращение в СРМ Розничный\nВыбирает\n\
94
+ Тематика: Банкротство физического лица\nПодтематика: Проведение операций\
95
+ \ по счету банкрота (в том числа открытие/закрытие)\nВыбирает причину обращения:\
96
+ \ РАЗБЛОКИРОВКА личного кабинета СБОЛ банкрота/ Блокировка Мобильного банка. \n\
97
+ В обращении обязательно указывает Фамилию Имя Отчество банкрота, День рождения.\
98
+ \ \nВкладывает в обращение:\n - Определение арбитражного суда о прекращении/завершении\
99
+ \ банкротства\n - ДУЛ Клиента \nОтправляет в работу."
100
+ - source_sentence: 'query: Кто может выступать в качестве инициатора изменения контракта
101
+ согласно документу с memo_id 63368?'
102
+ sentences:
103
+ - 'query: Кто имеет право запросить изменение контракта согласно документу с memo_id
104
+ 63368?'
105
+ - 'query: Когда можно изменить контракт согласно memo_id 63368?'
106
+ - 'query: Что необходимо отправить в ПЦП Центр комплаенс при подозрении на отмывание
107
+ денег?'
108
+ - source_sentence: 'query: Что нужно сделать, если клиент найден?'
109
+ sentences:
110
+ - 'query: Что делать, если произошел сбой в системе Сбол.про?'
111
+ - 'query: Какие шаги необходимо предпринять, если клиент обнаружен?'
112
+ - 'query: Что делать после того, как клиент найден?'
113
+ - source_sentence: 'query: Кто отвечает за подтверждение замены владельца номинального
114
+ счета?'
115
+ sentences:
116
+ - 'query: Кто должен дать согласие на смену владельца номинального счета?'
117
+ - 'query: Где можно установить индивидуальные лимиты по Детской СберКарте для законного
118
+ представителя?'
119
+ - 'query: Какой документ требуется для подтверждения личности клиента при смене
120
+ владельца номинального счета?'
121
+ pipeline_tag: sentence-similarity
122
+ library_name: sentence-transformers
123
+ ---
124
+
125
+ # SentenceTransformer based on intfloat/multilingual-e5-large
126
+
127
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [intfloat/multilingual-e5-large](https://huggingface.co/intfloat/multilingual-e5-large) on the q2q_data and q2p_data datasets. It maps sentences & paragraphs to a 1024-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
128
+
129
+ ## Model Details
130
+
131
+ ### Model Description
132
+ - **Model Type:** Sentence Transformer
133
+ - **Base model:** [intfloat/multilingual-e5-large](https://huggingface.co/intfloat/multilingual-e5-large) <!-- at revision 0dc5580a448e4284468b8909bae50fa925907bc5 -->
134
+ - **Maximum Sequence Length:** 512 tokens
135
+ - **Output Dimensionality:** 1024 dimensions
136
+ - **Similarity Function:** Cosine Similarity
137
+ - **Training Datasets:**
138
+ - q2q_data
139
+ - q2p_data
140
+ <!-- - **Language:** Unknown -->
141
+ <!-- - **License:** Unknown -->
142
+
143
+ ### Model Sources
144
+
145
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
146
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
147
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
148
+
149
+ ### Full Model Architecture
150
+
151
+ ```
152
+ SentenceTransformer(
153
+ (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: XLMRobertaModel
154
+ (1): Pooling({'word_embedding_dimension': 1024, '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})
155
+ (2): Normalize()
156
+ )
157
+ ```
158
+
159
+ ## Usage
160
+
161
+ ### Direct Usage (Sentence Transformers)
162
+
163
+ First install the Sentence Transformers library:
164
+
165
+ ```bash
166
+ pip install -U sentence-transformers
167
+ ```
168
+
169
+ Then you can load this model and run inference.
170
+ ```python
171
+ from sentence_transformers import SentenceTransformer
172
+
173
+ # Download from the 🤗 Hub
174
+ model = SentenceTransformer("George2002/sledopyt_embedder")
175
+ # Run inference
176
+ sentences = [
177
+ 'query: Кто отвечает за подтверждение замены владельца номинального счета?',
178
+ 'query: Кто должен дать согласие на смену владельца номинального счета?',
179
+ 'query: Какой документ требуется для подтверждения личности клиента при смене владельца номинального счета?',
180
+ ]
181
+ embeddings = model.encode(sentences)
182
+ print(embeddings.shape)
183
+ # [3, 1024]
184
+
185
+ # Get the similarity scores for the embeddings
186
+ similarities = model.similarity(embeddings, embeddings)
187
+ print(similarities.shape)
188
+ # [3, 3]
189
+ ```
190
+
191
+ <!--
192
+ ### Direct Usage (Transformers)
193
+
194
+ <details><summary>Click to see the direct usage in Transformers</summary>
195
+
196
+ </details>
197
+ -->
198
+
199
+ <!--
200
+ ### Downstream Usage (Sentence Transformers)
201
+
202
+ You can finetune this model on your own dataset.
203
+
204
+ <details><summary>Click to expand</summary>
205
+
206
+ </details>
207
+ -->
208
+
209
+ <!--
210
+ ### Out-of-Scope Use
211
+
212
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
213
+ -->
214
+
215
+ <!--
216
+ ## Bias, Risks and Limitations
217
+
218
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
219
+ -->
220
+
221
+ <!--
222
+ ### Recommendations
223
+
224
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
225
+ -->
226
+
227
+ ## Training Details
228
+
229
+ ### Training Datasets
230
+
231
+ #### q2q_data
232
+
233
+ * Dataset: q2q_data
234
+ * Size: 33,406 training samples
235
+ * Columns: <code>query_1</code>, <code>query_2</code>, and <code>negative</code>
236
+ * Approximate statistics based on the first 1000 samples:
237
+ | | query_1 | query_2 | negative |
238
+ |:--------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|
239
+ | type | string | string | string |
240
+ | details | <ul><li>min: 12 tokens</li><li>mean: 22.08 tokens</li><li>max: 42 tokens</li></ul> | <ul><li>min: 11 tokens</li><li>mean: 21.77 tokens</li><li>max: 43 tokens</li></ul> | <ul><li>min: 11 tokens</li><li>mean: 21.45 tokens</li><li>max: 39 tokens</li></ul> |
241
+ * Samples:
242
+ | query_1 | query_2 | negative |
243
+ |:-----------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------|
244
+ | <code>query: Какой телефон для отключения дневного расходного лимита для ребенка?</code> | <code>query: На какой номер нужно позвонить, чтобы снять лимит расходов на день для ребенка?</code> | <code>query: Как отключить лимиты по детской карте через мобильное приложение, если родитель хочет это сделать?</code> |
245
+ | <code>query: При каком условии можно выбрать возраст ребенка представительством?</code> | <code>query: При каких обстоятельствах можно определить возраст ребенка в представительстве?</code> | <code>query: Какой статус должен быть у представителя для определения возраста ребенка?</code> |
246
+ | <code>query: Какие бумаги нужны при обращении, если счет не удается найти?</code> | <code>query: Что требуется приложить к запросу, если счет отсутствует?</code> | <code>query: Что нужно предоставить для регистрации обращения, если счёт не обнаружен?</code> |
247
+ * Loss: [<code>TripletLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#tripletloss) with these parameters:
248
+ ```json
249
+ {
250
+ "distance_metric": "TripletDistanceMetric.EUCLIDEAN",
251
+ "triplet_margin": 5
252
+ }
253
+ ```
254
+
255
+ #### q2p_data
256
+
257
+ * Dataset: q2p_data
258
+ * Size: 7,709 training samples
259
+ * Columns: <code>query</code>, <code>chunk</code>, and <code>negative</code>
260
+ * Approximate statistics based on the first 1000 samples:
261
+ | | query | chunk | negative |
262
+ |:--------|:-----------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
263
+ | type | string | string | string |
264
+ | details | <ul><li>min: 13 tokens</li><li>mean: 22.14 tokens</li><li>max: 40 tokens</li></ul> | <ul><li>min: 12 tokens</li><li>mean: 162.33 tokens</li><li>max: 512 tokens</li></ul> | <ul><li>min: 12 tokens</li><li>mean: 129.72 tokens</li><li>max: 512 tokens</li></ul> |
265
+ * Samples:
266
+ | query | chunk | negative |
267
+ |:-----------------------------------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
268
+ | <code>query: Что такое ДУЛ законного представителя и зачем он нужен?</code> | <code>passage: Документы, необходимые для прикрепления законного представителя к подопечному (несовершеннолетнему от 14 до 18 лет):<br><br>Приемный родитель<br><br>ДУЛ законного представителя<br><br>Документ из органа опеки и попечительства о передаче ребенка в приемную семью</code> | <code>passage: Документы, необходимые для прикрепления законного представителя к подопечному (несовершеннолетнему от 14 до 18 лет):<br><br>Попечитель (физ. лицо)<br><br>ДУЛ законного представителя<br><br>Решение органа опеки и попечительства о назначении попечителя</code> |
269
+ | <code>query: Что требуется для получения выписки по счету вкладчика с ограниченной дееспособностью?</code> | <code>passage: Выписка по счету (вкладу) может быть выдана:<br><br>- Ограниченно дееспособный в силу пристрастия<br><br>-Доверенное лицо ограниченно дееспособного</code> | <code>passage: Кто обратился?<br><br>Вкладчик/Представитель вкладчика (полномочия подтверждены)<br><br>Выберите причину розыска?</code> |
270
+ | <code>query: Какие учредительные документы необходимы для государственного социального учреждения?</code> | <code>passage: Документ (распоряжение/приказ/акт/путевка) органа опеки и попечительства о помещении (передаче под надзор) несовершеннолетнего в государственную социальную организацию <br><br>Учредительные документы государственного социального учреждения (далее –учреждение)<br><br>Свидетельство о постановке на учет в налоговом органе учреждения<br><br>Выписка из Единого государственного реестра юридических лиц<br><br>Свидетельство о государственной регистрации учреждения<br><br>Документ (как правило, приказ) о назначении (избрании) единоличного исполнительного органа (директора) социального учреждения</code> | <code>passage: Документ (распоряжение/приказ/акт/путевка) органа опеки и попечительства о помещении (передаче под надзор) ограниченно дееспособного гражданина в государственную социальную организацию <br>Учредительные документы государственного социального учреждения (далее –учреждение)<br><br>Свидетельство о постановке на учет в налоговом органе учреждения<br><br>Выписка из Единого государственного реестра юридических лиц<br><br>Свидетельство о государственной регистрации учреждения<br><br>Документы об избрании (назначении) единоличного исполнительного органа</code> |
271
+ * Loss: [<code>TripletLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#tripletloss) with these parameters:
272
+ ```json
273
+ {
274
+ "distance_metric": "TripletDistanceMetric.EUCLIDEAN",
275
+ "triplet_margin": 5
276
+ }
277
+ ```
278
+
279
+ ### Evaluation Datasets
280
+
281
+ #### q2q_data
282
+
283
+ * Dataset: q2q_data
284
+ * Size: 1,759 evaluation samples
285
+ * Columns: <code>query_1</code>, <code>query_2</code>, and <code>negative</code>
286
+ * Approximate statistics based on the first 1000 samples:
287
+ | | query_1 | query_2 | negative |
288
+ |:--------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|
289
+ | type | string | string | string |
290
+ | details | <ul><li>min: 11 tokens</li><li>mean: 22.02 tokens</li><li>max: 40 tokens</li></ul> | <ul><li>min: 11 tokens</li><li>mean: 21.69 tokens</li><li>max: 39 tokens</li></ul> | <ul><li>min: 12 tokens</li><li>mean: 21.38 tokens</li><li>max: 37 tokens</li></ul> |
291
+ * Samples:
292
+ | query_1 | query_2 | negative |
293
+ |:---------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------|
294
+ | <code>query: Что необходимо сделать при получении отрицательного заключения?</code> | <code>query: Какие действия нужно предпринять после получения отрицательного ответа?</code> | <code>query: Какие шаги следует предпринять при получении негативного ответа после открытия счета?</code> |
295
+ | <code>query: Нужно ли согласовывать операцию с комплаенсом при переводе в СБОЛ.Про?</code> | <code>query: Обязательно ли получить согласие комплаенса для операции перевода в СБОЛ.Про?</code> | <code>query: Необходимо ли согласовывать перевод с комплаенсом в СБОЛ.Про?</code> |
296
+ | <code>query: Что появляется на экране после нажатия кнопки «Ребенок до 14 лет» или «Ребенок от 14 до 18 лет»?</code> | <code>query: Какой экран показывается после нажатия на кнопки «Ребенок до 14 лет» и «Ребенок от 14 до 18 лет»?</code> | <code>query: Каким лицам доступна выписка по счету несовершеннолетнего в возрасте от 14 до 18 лет?</code> |
297
+ * Loss: [<code>TripletLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#tripletloss) with these parameters:
298
+ ```json
299
+ {
300
+ "distance_metric": "TripletDistanceMetric.EUCLIDEAN",
301
+ "triplet_margin": 5
302
+ }
303
+ ```
304
+
305
+ #### q2p_data
306
+
307
+ * Dataset: q2p_data
308
+ * Size: 406 evaluation samples
309
+ * Columns: <code>query</code>, <code>chunk</code>, and <code>negative</code>
310
+ * Approximate statistics based on the first 406 samples:
311
+ | | query | chunk | negative |
312
+ |:--------|:-----------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
313
+ | type | string | string | string |
314
+ | details | <ul><li>min: 13 tokens</li><li>mean: 22.26 tokens</li><li>max: 40 tokens</li></ul> | <ul><li>min: 12 tokens</li><li>mean: 163.57 tokens</li><li>max: 512 tokens</li></ul> | <ul><li>min: 12 tokens</li><li>mean: 126.02 tokens</li><li>max: 512 tokens</li></ul> |
315
+ * Samples:
316
+ | query | chunk | negative |
317
+ |:----------------------------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
318
+ | <code>query: Куда следует направить информацию, полученную для розыска?</code> | <code>passage: Выберите причину розыска?<br><br>Информация нужна для предоставления в Суд/в органы опеки/ др. гос органы<br><br>Зарегистрируй обращение в СРМ «Розничный». Приложи (при наличии) документы, подтверждающие .....</code> | <code>passage: Заполните информацию о подопечном</code> |
319
+ | <code>query: Кто может обратиться за открытием карты МИР без согласия Финансового управляющего?</code> | <code>passage: С каким вопросом обратился банкрот?<br><br>06. Открытие счета/вклада (отличного от Специального счета банкрота)<br><br>Банкрот в любой стадии банкротства может открыть самостоятельно в стандартном режиме без согласия Финансового управляющего следующие виды счетов: <br><br>- Номинальный счет для получения пособий на подопечного/ алиментов на ребенка без согласия своего финансового управляющего. <br><br>Любой другой счет может открыть только банкрот в стадии реструктуризация долгов, предоставив разрешение Финансового управляющего<br><br>!!!!!! при обращении МОБИЛИЗОВАННОГО (участник СВО) банкрота (в стадии РИ) для открытия и получения карты МИР для зачисления военного довольствия, открываем и выдаем карту МИР БЕЗ РАЗРЕШЕНИЯ ФУ</code> | <code>passage: Банкрот в любой стадии банкротства может открыть самостоятельно в стандартном режиме без согласия Финансового управляющего следующие виды счетов: <br><br>- Номинальный счет для получения пособий на подопечного/ алиментов на ребенка без согласия своего финансового управляющего. <br><br>Любой другой счет может открыть только банкрот в стадии реструктуризация долгов, предоставив разрешение Финансового управляющего<br><br>!!!!!! при обращении МОБИЛИЗОВАННОГО (участник СВО) банкрота (в стадии РИ) для открытия и получения карты МИР для зачисления военного довольствия, открываем и выдаем карту МИР БЕЗ РАЗРЕШЕНИЯ ФУ<br><br>Процедура банкротства Реструктуризация долгов<br><br>Клиент, признанный банкротом в стадии реструктуризация долгов имеет право без согласия Финансового управляющего открыть счета:<br>- Специальный счет банкрота с ежемесячным лимитом в 50 000.00<br>- Счет ГЖС (Счет ГЖС открывается в стандартном режиме, арест/ блокировка на счет ГЖС не накладывается). <br>При открытии любого другого счёта банкрот должен п...</code> |
320
+ | <code>query: Что должно быть на свидетельстве о рождении Украины для оформления Детской СберКарты?</code> | <code>passage: ВНИМАНИЕ !<br>С 23 октября 2023г. в МП СБОЛ станет доступным заказ Детской СберКарты по свидетельству о рождении Украины со штампом о гражданстве РФ.<br>Для получения Детской СберКарты в офисе банка необходимо будет предоставить нотариально заверенное удостоверение перевода на русский язык свидетельства о рождении Украины со штампом о гражданстве РФ.<br><br><br><br><br><br>Типичные вопросы по Детской СберКарте и ответы на них<br><br>18. Сколько изготавливается Детская СберКарта ?<br><br>Актуальные сроки выпуска карты размещены на сайте Банка: Детская СберКарт -> Тарифы -> Документы и ссылки</code> | <code>passage: ВНИМАНИЕ !<br>С 23 октября 2023г. в МП СБОЛ станет доступным заказ Детской СберКарты по свидетельству о рождении Украины со штампом о гражданстве РФ.<br>Для получения Детской СберКарты в офисе банка необходимо будет предоставить нотариально заверенное удостоверение перевода на русский язык свидетельства о рождении Украины со штампом о гражданстве РФ.<br><br><br><br><br><br>Типичные вопросы по Детской СберКарте и ответы на них<br><br>05. Бонусы Спасибо по Детской СберКарте<br><br>Выберите интересующий вопрос</code> |
321
+ * Loss: [<code>TripletLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#tripletloss) with these parameters:
322
+ ```json
323
+ {
324
+ "distance_metric": "TripletDistanceMetric.EUCLIDEAN",
325
+ "triplet_margin": 5
326
+ }
327
+ ```
328
+
329
+ ### Training Hyperparameters
330
+ #### Non-Default Hyperparameters
331
+
332
+ - `eval_strategy`: steps
333
+ - `per_device_train_batch_size`: 36
334
+ - `learning_rate`: 1e-05
335
+ - `weight_decay`: 0.01
336
+ - `num_train_epochs`: 2
337
+ - `warmup_ratio`: 0.1
338
+ - `load_best_model_at_end`: True
339
+ - `push_to_hub`: True
340
+ - `hub_model_id`: George2002/sledopyt_embedder
341
+ - `hub_strategy`: end
342
+
343
+ #### All Hyperparameters
344
+ <details><summary>Click to expand</summary>
345
+
346
+ - `overwrite_output_dir`: False
347
+ - `do_predict`: False
348
+ - `eval_strategy`: steps
349
+ - `prediction_loss_only`: True
350
+ - `per_device_train_batch_size`: 36
351
+ - `per_device_eval_batch_size`: 8
352
+ - `per_gpu_train_batch_size`: None
353
+ - `per_gpu_eval_batch_size`: None
354
+ - `gradient_accumulation_steps`: 1
355
+ - `eval_accumulation_steps`: None
356
+ - `torch_empty_cache_steps`: None
357
+ - `learning_rate`: 1e-05
358
+ - `weight_decay`: 0.01
359
+ - `adam_beta1`: 0.9
360
+ - `adam_beta2`: 0.999
361
+ - `adam_epsilon`: 1e-08
362
+ - `max_grad_norm`: 1.0
363
+ - `num_train_epochs`: 2
364
+ - `max_steps`: -1
365
+ - `lr_scheduler_type`: linear
366
+ - `lr_scheduler_kwargs`: {}
367
+ - `warmup_ratio`: 0.1
368
+ - `warmup_steps`: 0
369
+ - `log_level`: passive
370
+ - `log_level_replica`: warning
371
+ - `log_on_each_node`: True
372
+ - `logging_nan_inf_filter`: True
373
+ - `save_safetensors`: True
374
+ - `save_on_each_node`: False
375
+ - `save_only_model`: False
376
+ - `restore_callback_states_from_checkpoint`: False
377
+ - `no_cuda`: False
378
+ - `use_cpu`: False
379
+ - `use_mps_device`: False
380
+ - `seed`: 42
381
+ - `data_seed`: None
382
+ - `jit_mode_eval`: False
383
+ - `use_ipex`: False
384
+ - `bf16`: False
385
+ - `fp16`: False
386
+ - `fp16_opt_level`: O1
387
+ - `half_precision_backend`: auto
388
+ - `bf16_full_eval`: False
389
+ - `fp16_full_eval`: False
390
+ - `tf32`: None
391
+ - `local_rank`: 0
392
+ - `ddp_backend`: None
393
+ - `tpu_num_cores`: None
394
+ - `tpu_metrics_debug`: False
395
+ - `debug`: []
396
+ - `dataloader_drop_last`: True
397
+ - `dataloader_num_workers`: 0
398
+ - `dataloader_prefetch_factor`: None
399
+ - `past_index`: -1
400
+ - `disable_tqdm`: False
401
+ - `remove_unused_columns`: True
402
+ - `label_names`: None
403
+ - `load_best_model_at_end`: True
404
+ - `ignore_data_skip`: False
405
+ - `fsdp`: []
406
+ - `fsdp_min_num_params`: 0
407
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
408
+ - `tp_size`: 0
409
+ - `fsdp_transformer_layer_cls_to_wrap`: None
410
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
411
+ - `deepspeed`: None
412
+ - `label_smoothing_factor`: 0.0
413
+ - `optim`: adamw_torch
414
+ - `optim_args`: None
415
+ - `adafactor`: False
416
+ - `group_by_length`: False
417
+ - `length_column_name`: length
418
+ - `ddp_find_unused_parameters`: None
419
+ - `ddp_bucket_cap_mb`: None
420
+ - `ddp_broadcast_buffers`: False
421
+ - `dataloader_pin_memory`: True
422
+ - `dataloader_persistent_workers`: False
423
+ - `skip_memory_metrics`: True
424
+ - `use_legacy_prediction_loop`: False
425
+ - `push_to_hub`: True
426
+ - `resume_from_checkpoint`: None
427
+ - `hub_model_id`: George2002/sledopyt_embedder
428
+ - `hub_strategy`: end
429
+ - `hub_private_repo`: None
430
+ - `hub_always_push`: False
431
+ - `gradient_checkpointing`: False
432
+ - `gradient_checkpointing_kwargs`: None
433
+ - `include_inputs_for_metrics`: False
434
+ - `include_for_metrics`: []
435
+ - `eval_do_concat_batches`: True
436
+ - `fp16_backend`: auto
437
+ - `push_to_hub_model_id`: None
438
+ - `push_to_hub_organization`: None
439
+ - `mp_parameters`:
440
+ - `auto_find_batch_size`: False
441
+ - `full_determinism`: False
442
+ - `torchdynamo`: None
443
+ - `ray_scope`: last
444
+ - `ddp_timeout`: 1800
445
+ - `torch_compile`: False
446
+ - `torch_compile_backend`: None
447
+ - `torch_compile_mode`: None
448
+ - `include_tokens_per_second`: False
449
+ - `include_num_input_tokens_seen`: False
450
+ - `neftune_noise_alpha`: None
451
+ - `optim_target_modules`: None
452
+ - `batch_eval_metrics`: False
453
+ - `eval_on_start`: False
454
+ - `use_liger_kernel`: False
455
+ - `eval_use_gather_object`: False
456
+ - `average_tokens_across_devices`: False
457
+ - `prompts`: None
458
+ - `batch_sampler`: batch_sampler
459
+ - `multi_dataset_batch_sampler`: proportional
460
+
461
+ </details>
462
+
463
+ ### Training Logs
464
+ <details><summary>Click to expand</summary>
465
+
466
+ | Epoch | Step | Training Loss | q2q data loss | q2p data loss |
467
+ |:------:|:----:|:-------------:|:-------------:|:-------------:|
468
+ | 0.0175 | 10 | 4.957 | - | - |
469
+ | 0.0351 | 20 | 4.9467 | - | - |
470
+ | 0.0526 | 30 | 4.9452 | - | - |
471
+ | 0.0702 | 40 | 4.9325 | - | - |
472
+ | 0.0877 | 50 | 4.9056 | 4.8804 | 4.9222 |
473
+ | 0.1053 | 60 | 4.9041 | - | - |
474
+ | 0.1228 | 70 | 4.8866 | - | - |
475
+ | 0.1404 | 80 | 4.844 | - | - |
476
+ | 0.1579 | 90 | 4.8064 | - | - |
477
+ | 0.1754 | 100 | 4.8182 | 4.7243 | 4.8660 |
478
+ | 0.1930 | 110 | 4.7791 | - | - |
479
+ | 0.2105 | 120 | 4.7659 | - | - |
480
+ | 0.2281 | 130 | 4.7572 | - | - |
481
+ | 0.2456 | 140 | 4.7234 | - | - |
482
+ | 0.2632 | 150 | 4.726 | 4.6268 | 4.8712 |
483
+ | 0.2807 | 160 | 4.6932 | - | - |
484
+ | 0.2982 | 170 | 4.6654 | - | - |
485
+ | 0.3158 | 180 | 4.6776 | - | - |
486
+ | 0.3333 | 190 | 4.6617 | - | - |
487
+ | 0.3509 | 200 | 4.6928 | 4.5581 | 4.8814 |
488
+ | 0.3684 | 210 | 4.6497 | - | - |
489
+ | 0.3860 | 220 | 4.677 | - | - |
490
+ | 0.4035 | 230 | 4.6344 | - | - |
491
+ | 0.4211 | 240 | 4.6612 | - | - |
492
+ | 0.4386 | 250 | 4.6274 | 4.5154 | 4.8396 |
493
+ | 0.4561 | 260 | 4.6556 | - | - |
494
+ | 0.4737 | 270 | 4.6382 | - | - |
495
+ | 0.4912 | 280 | 4.6053 | - | - |
496
+ | 0.5088 | 290 | 4.6131 | - | - |
497
+ | 0.5263 | 300 | 4.6453 | 4.4957 | 4.8314 |
498
+ | 0.5439 | 310 | 4.5819 | - | - |
499
+ | 0.5614 | 320 | 4.5948 | - | - |
500
+ | 0.5789 | 330 | 4.5288 | - | - |
501
+ | 0.5965 | 340 | 4.6152 | - | - |
502
+ | 0.6140 | 350 | 4.5831 | 4.4657 | 4.7953 |
503
+ | 0.6316 | 360 | 4.5507 | - | - |
504
+ | 0.6491 | 370 | 4.5718 | - | - |
505
+ | 0.6667 | 380 | 4.6269 | - | - |
506
+ | 0.6842 | 390 | 4.6017 | - | - |
507
+ | 0.7018 | 400 | 4.5155 | 4.4396 | 4.7694 |
508
+ | 0.7193 | 410 | 4.5055 | - | - |
509
+ | 0.7368 | 420 | 4.534 | - | - |
510
+ | 0.7544 | 430 | 4.5358 | - | - |
511
+ | 0.7719 | 440 | 4.5443 | - | - |
512
+ | 0.7895 | 450 | 4.5309 | 4.4183 | 4.7751 |
513
+ | 0.8070 | 460 | 4.5952 | - | - |
514
+ | 0.8246 | 470 | 4.5561 | - | - |
515
+ | 0.8421 | 480 | 4.5191 | - | - |
516
+ | 0.8596 | 490 | 4.5066 | - | - |
517
+ | 0.8772 | 500 | 4.4875 | 4.4138 | 4.8195 |
518
+ | 0.8947 | 510 | 4.5051 | - | - |
519
+ | 0.9123 | 520 | 4.4872 | - | - |
520
+ | 0.9298 | 530 | 4.4918 | - | - |
521
+ | 0.9474 | 540 | 4.5357 | - | - |
522
+ | 0.9649 | 550 | 4.4898 | 4.3754 | 4.7799 |
523
+ | 0.9825 | 560 | 4.5742 | - | - |
524
+ | 1.0 | 570 | 4.5461 | - | - |
525
+ | 1.0175 | 580 | 4.5505 | - | - |
526
+ | 1.0351 | 590 | 4.5027 | - | - |
527
+ | 1.0526 | 600 | 4.5747 | 4.4060 | 4.7915 |
528
+ | 1.0702 | 610 | 4.5296 | - | - |
529
+ | 1.0877 | 620 | 4.4262 | - | - |
530
+ | 1.1053 | 630 | 4.5415 | - | - |
531
+ | 1.1228 | 640 | 4.5386 | - | - |
532
+ | 1.1404 | 650 | 4.4552 | 4.3632 | 4.8105 |
533
+ | 1.1579 | 660 | 4.4473 | - | - |
534
+ | 1.1754 | 670 | 4.5069 | - | - |
535
+ | 1.1930 | 680 | 4.5129 | - | - |
536
+ | 1.2105 | 690 | 4.4611 | - | - |
537
+ | 1.2281 | 700 | 4.5104 | 4.3530 | 4.7875 |
538
+ | 1.2456 | 710 | 4.4742 | - | - |
539
+ | 1.2632 | 720 | 4.4887 | - | - |
540
+ | 1.2807 | 730 | 4.406 | - | - |
541
+ | 1.2982 | 740 | 4.4049 | - | - |
542
+ | 1.3158 | 750 | 4.4165 | 4.3484 | 4.7866 |
543
+ | 1.3333 | 760 | 4.4274 | - | - |
544
+ | 1.3509 | 770 | 4.4855 | - | - |
545
+ | 1.3684 | 780 | 4.4571 | - | - |
546
+ | 1.3860 | 790 | 4.4307 | - | - |
547
+ | 1.4035 | 800 | 4.4387 | 4.3450 | 4.7628 |
548
+ | 1.4211 | 810 | 4.4592 | - | - |
549
+ | 1.4386 | 820 | 4.4368 | - | - |
550
+ | 1.4561 | 830 | 4.4863 | - | - |
551
+ | 1.4737 | 840 | 4.463 | - | - |
552
+ | 1.4912 | 850 | 4.4113 | 4.3252 | 4.7610 |
553
+ | 1.5088 | 860 | 4.4368 | - | - |
554
+ | 1.5263 | 870 | 4.4738 | - | - |
555
+ | 1.5439 | 880 | 4.4195 | - | - |
556
+ | 1.5614 | 890 | 4.4478 | - | - |
557
+ | 1.5789 | 900 | 4.3849 | 4.3140 | 4.7519 |
558
+ | 1.5965 | 910 | 4.4896 | - | - |
559
+ | 1.6140 | 920 | 4.4301 | - | - |
560
+ | 1.6316 | 930 | 4.4142 | - | - |
561
+ | 1.6491 | 940 | 4.4582 | - | - |
562
+ | 1.6667 | 950 | 4.5075 | 4.3189 | 4.7259 |
563
+ | 1.6842 | 960 | 4.4454 | - | - |
564
+ | 1.7018 | 970 | 4.3547 | - | - |
565
+ | 1.7193 | 980 | 4.4016 | - | - |
566
+ | 1.7368 | 990 | 4.4064 | - | - |
567
+ | 1.7544 | 1000 | 4.4356 | 4.3151 | 4.7276 |
568
+ | 1.7719 | 1010 | 4.4105 | - | - |
569
+ | 1.7895 | 1020 | 4.4067 | - | - |
570
+ | 1.8070 | 1030 | 4.4296 | - | - |
571
+ | 1.8246 | 1040 | 4.4147 | - | - |
572
+ | 1.8421 | 1050 | 4.3743 | 4.3136 | 4.7182 |
573
+ | 1.8596 | 1060 | 4.4065 | - | - |
574
+ | 1.8772 | 1070 | 4.4025 | - | - |
575
+ | 1.8947 | 1080 | 4.3912 | - | - |
576
+ | 1.9123 | 1090 | 4.3731 | - | - |
577
+ | 1.9298 | 1100 | 4.3817 | 4.3120 | 4.7357 |
578
+ | 1.9474 | 1110 | 4.4305 | - | - |
579
+ | 1.9649 | 1120 | 4.3914 | - | - |
580
+ | 1.9825 | 1130 | 4.4753 | - | - |
581
+ | 2.0 | 1140 | 4.4536 | - | - |
582
+
583
+ </details>
584
+
585
+ ### Framework Versions
586
+ - Python: 3.10.12
587
+ - Sentence Transformers: 4.1.0
588
+ - Transformers: 4.51.3
589
+ - PyTorch: 2.6.0+cu124
590
+ - Accelerate: 1.6.0
591
+ - Datasets: 3.5.0
592
+ - Tokenizers: 0.21.1
593
+
594
+ ## Citation
595
+
596
+ ### BibTeX
597
+
598
+ #### Sentence Transformers
599
+ ```bibtex
600
+ @inproceedings{reimers-2019-sentence-bert,
601
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
602
+ author = "Reimers, Nils and Gurevych, Iryna",
603
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
604
+ month = "11",
605
+ year = "2019",
606
+ publisher = "Association for Computational Linguistics",
607
+ url = "https://arxiv.org/abs/1908.10084",
608
+ }
609
+ ```
610
+
611
+ #### TripletLoss
612
+ ```bibtex
613
+ @misc{hermans2017defense,
614
+ title={In Defense of the Triplet Loss for Person Re-Identification},
615
+ author={Alexander Hermans and Lucas Beyer and Bastian Leibe},
616
+ year={2017},
617
+ eprint={1703.07737},
618
+ archivePrefix={arXiv},
619
+ primaryClass={cs.CV}
620
+ }
621
+ ```
622
+
623
+ <!--
624
+ ## Glossary
625
+
626
+ *Clearly define terms in order to be accessible across audiences.*
627
+ -->
628
+
629
+ <!--
630
+ ## Model Card Authors
631
+
632
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
633
+ -->
634
+
635
+ <!--
636
+ ## Model Card Contact
637
+
638
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
639
+ -->
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "XLMRobertaModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": null,
8
+ "eos_token_id": 2,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 1024,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 4096,
14
+ "layer_norm_eps": 1e-05,
15
+ "max_position_embeddings": 514,
16
+ "model_type": "xlm-roberta",
17
+ "num_attention_heads": 16,
18
+ "num_hidden_layers": 24,
19
+ "output_past": true,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "torch_dtype": "float32",
23
+ "transformers_version": "4.51.3",
24
+ "type_vocab_size": 1,
25
+ "use_cache": true,
26
+ "vocab_size": 250002
27
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "4.1.0",
4
+ "transformers": "4.51.3",
5
+ "pytorch": "2.6.0+cu124"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null,
9
+ "similarity_fn_name": "cosine"
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7515018f5affd45fd06031441cc6f80d56d3d6ace9f66b500dacd81c8dcaf6e
3
+ size 2239607176
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.models.Normalize"
19
+ }
20
+ ]
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e42702ecb4deb779013ae19516cda637b6cc7c753393745b81634d880bdc41f3
3
+ size 4471055801
rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:146f18108292a900e3a747e95127a2f108fe8808f83bef93e35c5d393b53ef2b
3
+ size 14512
rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f7478be6a7c1fb7a5d40ac5ae3c8d42f6d2436c0e5da02cf1a8695abfa2d412
3
+ size 14512
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f09c89c7de3256d9b3cbdea6949c3738ef7b72eea9145038ad05df05556b21c7
3
+ size 1064
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:883b037111086fd4dfebbbc9b7cee11e1517b5e0c0514879478661440f137085
3
+ size 17082987
tokenizer_config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "250001": {
36
+ "content": "<mask>",
37
+ "lstrip": true,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "<s>",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "<s>",
47
+ "eos_token": "</s>",
48
+ "extra_special_tokens": {},
49
+ "mask_token": "<mask>",
50
+ "model_max_length": 512,
51
+ "pad_token": "<pad>",
52
+ "sep_token": "</s>",
53
+ "tokenizer_class": "XLMRobertaTokenizer",
54
+ "unk_token": "<unk>"
55
+ }
trainer_state.json ADDED
@@ -0,0 +1,1184 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 1050,
3
+ "best_metric": 4.7181901931762695,
4
+ "best_model_checkpoint": ".../training_output/checkpoint-800",
5
+ "epoch": 2.0,
6
+ "eval_steps": 50,
7
+ "global_step": 1140,
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.017543859649122806,
14
+ "grad_norm": 0.8221026659011841,
15
+ "learning_rate": 7.894736842105263e-07,
16
+ "loss": 4.957,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.03508771929824561,
21
+ "grad_norm": 0.8544751405715942,
22
+ "learning_rate": 1.6666666666666667e-06,
23
+ "loss": 4.9467,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.05263157894736842,
28
+ "grad_norm": 0.964083731174469,
29
+ "learning_rate": 2.5438596491228075e-06,
30
+ "loss": 4.9452,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.07017543859649122,
35
+ "grad_norm": 0.9615139365196228,
36
+ "learning_rate": 3.421052631578948e-06,
37
+ "loss": 4.9325,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.08771929824561403,
42
+ "grad_norm": 1.156923770904541,
43
+ "learning_rate": 4.298245614035088e-06,
44
+ "loss": 4.9056,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.08771929824561403,
49
+ "eval_q2q_data_loss": 4.880394458770752,
50
+ "eval_q2q_data_runtime": 5.5966,
51
+ "eval_q2q_data_samples_per_second": 314.295,
52
+ "eval_q2q_data_steps_per_second": 19.655,
53
+ "step": 50
54
+ },
55
+ {
56
+ "epoch": 0.08771929824561403,
57
+ "eval_q2p_data_loss": 4.922183990478516,
58
+ "eval_q2p_data_runtime": 7.55,
59
+ "eval_q2p_data_samples_per_second": 53.775,
60
+ "eval_q2p_data_steps_per_second": 3.444,
61
+ "step": 50
62
+ },
63
+ {
64
+ "epoch": 0.10526315789473684,
65
+ "grad_norm": 1.2874988317489624,
66
+ "learning_rate": 5.175438596491229e-06,
67
+ "loss": 4.9041,
68
+ "step": 60
69
+ },
70
+ {
71
+ "epoch": 0.12280701754385964,
72
+ "grad_norm": 1.5450624227523804,
73
+ "learning_rate": 6.0526315789473685e-06,
74
+ "loss": 4.8866,
75
+ "step": 70
76
+ },
77
+ {
78
+ "epoch": 0.14035087719298245,
79
+ "grad_norm": 1.8990825414657593,
80
+ "learning_rate": 6.92982456140351e-06,
81
+ "loss": 4.844,
82
+ "step": 80
83
+ },
84
+ {
85
+ "epoch": 0.15789473684210525,
86
+ "grad_norm": 2.0947864055633545,
87
+ "learning_rate": 7.80701754385965e-06,
88
+ "loss": 4.8064,
89
+ "step": 90
90
+ },
91
+ {
92
+ "epoch": 0.17543859649122806,
93
+ "grad_norm": 2.2433862686157227,
94
+ "learning_rate": 8.68421052631579e-06,
95
+ "loss": 4.8182,
96
+ "step": 100
97
+ },
98
+ {
99
+ "epoch": 0.17543859649122806,
100
+ "eval_q2q_data_loss": 4.724327087402344,
101
+ "eval_q2q_data_runtime": 5.5749,
102
+ "eval_q2q_data_samples_per_second": 315.523,
103
+ "eval_q2q_data_steps_per_second": 19.731,
104
+ "step": 100
105
+ },
106
+ {
107
+ "epoch": 0.17543859649122806,
108
+ "eval_q2p_data_loss": 4.865963459014893,
109
+ "eval_q2p_data_runtime": 7.5397,
110
+ "eval_q2p_data_samples_per_second": 53.849,
111
+ "eval_q2p_data_steps_per_second": 3.448,
112
+ "step": 100
113
+ },
114
+ {
115
+ "epoch": 0.19298245614035087,
116
+ "grad_norm": 2.198146104812622,
117
+ "learning_rate": 9.56140350877193e-06,
118
+ "loss": 4.7791,
119
+ "step": 110
120
+ },
121
+ {
122
+ "epoch": 0.21052631578947367,
123
+ "grad_norm": 2.6786892414093018,
124
+ "learning_rate": 9.951267056530215e-06,
125
+ "loss": 4.7659,
126
+ "step": 120
127
+ },
128
+ {
129
+ "epoch": 0.22807017543859648,
130
+ "grad_norm": 2.485137462615967,
131
+ "learning_rate": 9.853801169590644e-06,
132
+ "loss": 4.7572,
133
+ "step": 130
134
+ },
135
+ {
136
+ "epoch": 0.24561403508771928,
137
+ "grad_norm": 2.5113883018493652,
138
+ "learning_rate": 9.756335282651072e-06,
139
+ "loss": 4.7234,
140
+ "step": 140
141
+ },
142
+ {
143
+ "epoch": 0.2631578947368421,
144
+ "grad_norm": 3.184298276901245,
145
+ "learning_rate": 9.658869395711503e-06,
146
+ "loss": 4.726,
147
+ "step": 150
148
+ },
149
+ {
150
+ "epoch": 0.2631578947368421,
151
+ "eval_q2q_data_loss": 4.626772403717041,
152
+ "eval_q2q_data_runtime": 5.5905,
153
+ "eval_q2q_data_samples_per_second": 314.638,
154
+ "eval_q2q_data_steps_per_second": 19.676,
155
+ "step": 150
156
+ },
157
+ {
158
+ "epoch": 0.2631578947368421,
159
+ "eval_q2p_data_loss": 4.871231555938721,
160
+ "eval_q2p_data_runtime": 7.5434,
161
+ "eval_q2p_data_samples_per_second": 53.822,
162
+ "eval_q2p_data_steps_per_second": 3.447,
163
+ "step": 150
164
+ },
165
+ {
166
+ "epoch": 0.2807017543859649,
167
+ "grad_norm": 3.1563026905059814,
168
+ "learning_rate": 9.56140350877193e-06,
169
+ "loss": 4.6932,
170
+ "step": 160
171
+ },
172
+ {
173
+ "epoch": 0.2982456140350877,
174
+ "grad_norm": 3.4077727794647217,
175
+ "learning_rate": 9.463937621832359e-06,
176
+ "loss": 4.6654,
177
+ "step": 170
178
+ },
179
+ {
180
+ "epoch": 0.3157894736842105,
181
+ "grad_norm": 3.617626428604126,
182
+ "learning_rate": 9.366471734892788e-06,
183
+ "loss": 4.6776,
184
+ "step": 180
185
+ },
186
+ {
187
+ "epoch": 0.3333333333333333,
188
+ "grad_norm": 4.701232433319092,
189
+ "learning_rate": 9.269005847953217e-06,
190
+ "loss": 4.6617,
191
+ "step": 190
192
+ },
193
+ {
194
+ "epoch": 0.3508771929824561,
195
+ "grad_norm": 7.48028564453125,
196
+ "learning_rate": 9.171539961013646e-06,
197
+ "loss": 4.6928,
198
+ "step": 200
199
+ },
200
+ {
201
+ "epoch": 0.3508771929824561,
202
+ "eval_q2q_data_loss": 4.558098793029785,
203
+ "eval_q2q_data_runtime": 5.5778,
204
+ "eval_q2q_data_samples_per_second": 315.355,
205
+ "eval_q2q_data_steps_per_second": 19.721,
206
+ "step": 200
207
+ },
208
+ {
209
+ "epoch": 0.3508771929824561,
210
+ "eval_q2p_data_loss": 4.881445407867432,
211
+ "eval_q2p_data_runtime": 7.5112,
212
+ "eval_q2p_data_samples_per_second": 54.053,
213
+ "eval_q2p_data_steps_per_second": 3.462,
214
+ "step": 200
215
+ },
216
+ {
217
+ "epoch": 0.3684210526315789,
218
+ "grad_norm": 4.592555522918701,
219
+ "learning_rate": 9.074074074074075e-06,
220
+ "loss": 4.6497,
221
+ "step": 210
222
+ },
223
+ {
224
+ "epoch": 0.38596491228070173,
225
+ "grad_norm": 4.758955478668213,
226
+ "learning_rate": 8.976608187134503e-06,
227
+ "loss": 4.677,
228
+ "step": 220
229
+ },
230
+ {
231
+ "epoch": 0.40350877192982454,
232
+ "grad_norm": 4.005542278289795,
233
+ "learning_rate": 8.879142300194934e-06,
234
+ "loss": 4.6344,
235
+ "step": 230
236
+ },
237
+ {
238
+ "epoch": 0.42105263157894735,
239
+ "grad_norm": 5.429654598236084,
240
+ "learning_rate": 8.781676413255361e-06,
241
+ "loss": 4.6612,
242
+ "step": 240
243
+ },
244
+ {
245
+ "epoch": 0.43859649122807015,
246
+ "grad_norm": 5.14253044128418,
247
+ "learning_rate": 8.68421052631579e-06,
248
+ "loss": 4.6274,
249
+ "step": 250
250
+ },
251
+ {
252
+ "epoch": 0.43859649122807015,
253
+ "eval_q2q_data_loss": 4.515370845794678,
254
+ "eval_q2q_data_runtime": 5.5777,
255
+ "eval_q2q_data_samples_per_second": 315.36,
256
+ "eval_q2q_data_steps_per_second": 19.721,
257
+ "step": 250
258
+ },
259
+ {
260
+ "epoch": 0.43859649122807015,
261
+ "eval_q2p_data_loss": 4.839608669281006,
262
+ "eval_q2p_data_runtime": 7.5286,
263
+ "eval_q2p_data_samples_per_second": 53.928,
264
+ "eval_q2p_data_steps_per_second": 3.454,
265
+ "step": 250
266
+ },
267
+ {
268
+ "epoch": 0.45614035087719296,
269
+ "grad_norm": 4.397937774658203,
270
+ "learning_rate": 8.586744639376219e-06,
271
+ "loss": 4.6556,
272
+ "step": 260
273
+ },
274
+ {
275
+ "epoch": 0.47368421052631576,
276
+ "grad_norm": 6.12044095993042,
277
+ "learning_rate": 8.489278752436648e-06,
278
+ "loss": 4.6382,
279
+ "step": 270
280
+ },
281
+ {
282
+ "epoch": 0.49122807017543857,
283
+ "grad_norm": 8.43116283416748,
284
+ "learning_rate": 8.391812865497077e-06,
285
+ "loss": 4.6053,
286
+ "step": 280
287
+ },
288
+ {
289
+ "epoch": 0.5087719298245614,
290
+ "grad_norm": 7.88032341003418,
291
+ "learning_rate": 8.294346978557506e-06,
292
+ "loss": 4.6131,
293
+ "step": 290
294
+ },
295
+ {
296
+ "epoch": 0.5263157894736842,
297
+ "grad_norm": 6.561196804046631,
298
+ "learning_rate": 8.196881091617934e-06,
299
+ "loss": 4.6453,
300
+ "step": 300
301
+ },
302
+ {
303
+ "epoch": 0.5263157894736842,
304
+ "eval_q2q_data_loss": 4.495702743530273,
305
+ "eval_q2q_data_runtime": 5.5691,
306
+ "eval_q2q_data_samples_per_second": 315.85,
307
+ "eval_q2q_data_steps_per_second": 19.752,
308
+ "step": 300
309
+ },
310
+ {
311
+ "epoch": 0.5263157894736842,
312
+ "eval_q2p_data_loss": 4.831414222717285,
313
+ "eval_q2p_data_runtime": 7.5076,
314
+ "eval_q2p_data_samples_per_second": 54.079,
315
+ "eval_q2p_data_steps_per_second": 3.463,
316
+ "step": 300
317
+ },
318
+ {
319
+ "epoch": 0.543859649122807,
320
+ "grad_norm": 7.7354536056518555,
321
+ "learning_rate": 8.099415204678363e-06,
322
+ "loss": 4.5819,
323
+ "step": 310
324
+ },
325
+ {
326
+ "epoch": 0.5614035087719298,
327
+ "grad_norm": 6.592026233673096,
328
+ "learning_rate": 8.001949317738792e-06,
329
+ "loss": 4.5948,
330
+ "step": 320
331
+ },
332
+ {
333
+ "epoch": 0.5789473684210527,
334
+ "grad_norm": 8.176568031311035,
335
+ "learning_rate": 7.904483430799221e-06,
336
+ "loss": 4.5288,
337
+ "step": 330
338
+ },
339
+ {
340
+ "epoch": 0.5964912280701754,
341
+ "grad_norm": 8.80689811706543,
342
+ "learning_rate": 7.80701754385965e-06,
343
+ "loss": 4.6152,
344
+ "step": 340
345
+ },
346
+ {
347
+ "epoch": 0.6140350877192983,
348
+ "grad_norm": 6.051924228668213,
349
+ "learning_rate": 7.70955165692008e-06,
350
+ "loss": 4.5831,
351
+ "step": 350
352
+ },
353
+ {
354
+ "epoch": 0.6140350877192983,
355
+ "eval_q2q_data_loss": 4.4657182693481445,
356
+ "eval_q2q_data_runtime": 5.5705,
357
+ "eval_q2q_data_samples_per_second": 315.77,
358
+ "eval_q2q_data_steps_per_second": 19.747,
359
+ "step": 350
360
+ },
361
+ {
362
+ "epoch": 0.6140350877192983,
363
+ "eval_q2p_data_loss": 4.795331001281738,
364
+ "eval_q2p_data_runtime": 7.5177,
365
+ "eval_q2p_data_samples_per_second": 54.006,
366
+ "eval_q2p_data_steps_per_second": 3.458,
367
+ "step": 350
368
+ },
369
+ {
370
+ "epoch": 0.631578947368421,
371
+ "grad_norm": 6.087244510650635,
372
+ "learning_rate": 7.612085769980507e-06,
373
+ "loss": 4.5507,
374
+ "step": 360
375
+ },
376
+ {
377
+ "epoch": 0.6491228070175439,
378
+ "grad_norm": 8.209424018859863,
379
+ "learning_rate": 7.5146198830409365e-06,
380
+ "loss": 4.5718,
381
+ "step": 370
382
+ },
383
+ {
384
+ "epoch": 0.6666666666666666,
385
+ "grad_norm": 11.899641990661621,
386
+ "learning_rate": 7.417153996101365e-06,
387
+ "loss": 4.6269,
388
+ "step": 380
389
+ },
390
+ {
391
+ "epoch": 0.6842105263157895,
392
+ "grad_norm": 10.490060806274414,
393
+ "learning_rate": 7.319688109161795e-06,
394
+ "loss": 4.6017,
395
+ "step": 390
396
+ },
397
+ {
398
+ "epoch": 0.7017543859649122,
399
+ "grad_norm": 6.545611381530762,
400
+ "learning_rate": 7.222222222222223e-06,
401
+ "loss": 4.5155,
402
+ "step": 400
403
+ },
404
+ {
405
+ "epoch": 0.7017543859649122,
406
+ "eval_q2q_data_loss": 4.439589500427246,
407
+ "eval_q2q_data_runtime": 5.563,
408
+ "eval_q2q_data_samples_per_second": 316.195,
409
+ "eval_q2q_data_steps_per_second": 19.773,
410
+ "step": 400
411
+ },
412
+ {
413
+ "epoch": 0.7017543859649122,
414
+ "eval_q2p_data_loss": 4.769360542297363,
415
+ "eval_q2p_data_runtime": 7.5013,
416
+ "eval_q2p_data_samples_per_second": 54.124,
417
+ "eval_q2p_data_steps_per_second": 3.466,
418
+ "step": 400
419
+ },
420
+ {
421
+ "epoch": 0.7192982456140351,
422
+ "grad_norm": 9.658538818359375,
423
+ "learning_rate": 7.124756335282652e-06,
424
+ "loss": 4.5055,
425
+ "step": 410
426
+ },
427
+ {
428
+ "epoch": 0.7368421052631579,
429
+ "grad_norm": 11.859044075012207,
430
+ "learning_rate": 7.02729044834308e-06,
431
+ "loss": 4.534,
432
+ "step": 420
433
+ },
434
+ {
435
+ "epoch": 0.7543859649122807,
436
+ "grad_norm": 6.311577320098877,
437
+ "learning_rate": 6.92982456140351e-06,
438
+ "loss": 4.5358,
439
+ "step": 430
440
+ },
441
+ {
442
+ "epoch": 0.7719298245614035,
443
+ "grad_norm": 15.303114891052246,
444
+ "learning_rate": 6.832358674463938e-06,
445
+ "loss": 4.5443,
446
+ "step": 440
447
+ },
448
+ {
449
+ "epoch": 0.7894736842105263,
450
+ "grad_norm": 7.770440101623535,
451
+ "learning_rate": 6.7348927875243675e-06,
452
+ "loss": 4.5309,
453
+ "step": 450
454
+ },
455
+ {
456
+ "epoch": 0.7894736842105263,
457
+ "eval_q2q_data_loss": 4.418254852294922,
458
+ "eval_q2q_data_runtime": 5.5809,
459
+ "eval_q2q_data_samples_per_second": 315.182,
460
+ "eval_q2q_data_steps_per_second": 19.71,
461
+ "step": 450
462
+ },
463
+ {
464
+ "epoch": 0.7894736842105263,
465
+ "eval_q2p_data_loss": 4.7750725746154785,
466
+ "eval_q2p_data_runtime": 7.5356,
467
+ "eval_q2p_data_samples_per_second": 53.878,
468
+ "eval_q2p_data_steps_per_second": 3.45,
469
+ "step": 450
470
+ },
471
+ {
472
+ "epoch": 0.8070175438596491,
473
+ "grad_norm": 10.787198066711426,
474
+ "learning_rate": 6.637426900584796e-06,
475
+ "loss": 4.5952,
476
+ "step": 460
477
+ },
478
+ {
479
+ "epoch": 0.8245614035087719,
480
+ "grad_norm": 6.622506141662598,
481
+ "learning_rate": 6.539961013645225e-06,
482
+ "loss": 4.5561,
483
+ "step": 470
484
+ },
485
+ {
486
+ "epoch": 0.8421052631578947,
487
+ "grad_norm": 9.452810287475586,
488
+ "learning_rate": 6.442495126705654e-06,
489
+ "loss": 4.5191,
490
+ "step": 480
491
+ },
492
+ {
493
+ "epoch": 0.8596491228070176,
494
+ "grad_norm": 8.921065330505371,
495
+ "learning_rate": 6.345029239766083e-06,
496
+ "loss": 4.5066,
497
+ "step": 490
498
+ },
499
+ {
500
+ "epoch": 0.8771929824561403,
501
+ "grad_norm": 6.36785364151001,
502
+ "learning_rate": 6.247563352826511e-06,
503
+ "loss": 4.4875,
504
+ "step": 500
505
+ },
506
+ {
507
+ "epoch": 0.8771929824561403,
508
+ "eval_q2q_data_loss": 4.413846015930176,
509
+ "eval_q2q_data_runtime": 5.5964,
510
+ "eval_q2q_data_samples_per_second": 314.308,
511
+ "eval_q2q_data_steps_per_second": 19.655,
512
+ "step": 500
513
+ },
514
+ {
515
+ "epoch": 0.8771929824561403,
516
+ "eval_q2p_data_loss": 4.819548606872559,
517
+ "eval_q2p_data_runtime": 7.5407,
518
+ "eval_q2p_data_samples_per_second": 53.841,
519
+ "eval_q2p_data_steps_per_second": 3.448,
520
+ "step": 500
521
+ },
522
+ {
523
+ "epoch": 0.8947368421052632,
524
+ "grad_norm": 8.613053321838379,
525
+ "learning_rate": 6.15009746588694e-06,
526
+ "loss": 4.5051,
527
+ "step": 510
528
+ },
529
+ {
530
+ "epoch": 0.9122807017543859,
531
+ "grad_norm": 6.249648571014404,
532
+ "learning_rate": 6.0526315789473685e-06,
533
+ "loss": 4.4872,
534
+ "step": 520
535
+ },
536
+ {
537
+ "epoch": 0.9298245614035088,
538
+ "grad_norm": 14.66945743560791,
539
+ "learning_rate": 5.9551656920077984e-06,
540
+ "loss": 4.4918,
541
+ "step": 530
542
+ },
543
+ {
544
+ "epoch": 0.9473684210526315,
545
+ "grad_norm": 13.305913925170898,
546
+ "learning_rate": 5.857699805068227e-06,
547
+ "loss": 4.5357,
548
+ "step": 540
549
+ },
550
+ {
551
+ "epoch": 0.9649122807017544,
552
+ "grad_norm": 10.659647941589355,
553
+ "learning_rate": 5.760233918128656e-06,
554
+ "loss": 4.4898,
555
+ "step": 550
556
+ },
557
+ {
558
+ "epoch": 0.9649122807017544,
559
+ "eval_q2q_data_loss": 4.375401020050049,
560
+ "eval_q2q_data_runtime": 5.5712,
561
+ "eval_q2q_data_samples_per_second": 315.731,
562
+ "eval_q2q_data_steps_per_second": 19.744,
563
+ "step": 550
564
+ },
565
+ {
566
+ "epoch": 0.9649122807017544,
567
+ "eval_q2p_data_loss": 4.779933929443359,
568
+ "eval_q2p_data_runtime": 7.4961,
569
+ "eval_q2p_data_samples_per_second": 54.162,
570
+ "eval_q2p_data_steps_per_second": 3.468,
571
+ "step": 550
572
+ },
573
+ {
574
+ "epoch": 0.9824561403508771,
575
+ "grad_norm": 7.730218410491943,
576
+ "learning_rate": 5.662768031189084e-06,
577
+ "loss": 4.5742,
578
+ "step": 560
579
+ },
580
+ {
581
+ "epoch": 1.0,
582
+ "grad_norm": 9.418205261230469,
583
+ "learning_rate": 5.565302144249514e-06,
584
+ "loss": 4.5461,
585
+ "step": 570
586
+ },
587
+ {
588
+ "epoch": 1.0175438596491229,
589
+ "grad_norm": 10.373188972473145,
590
+ "learning_rate": 5.467836257309942e-06,
591
+ "loss": 4.5505,
592
+ "step": 580
593
+ },
594
+ {
595
+ "epoch": 1.0350877192982457,
596
+ "grad_norm": 11.559415817260742,
597
+ "learning_rate": 5.370370370370371e-06,
598
+ "loss": 4.5027,
599
+ "step": 590
600
+ },
601
+ {
602
+ "epoch": 1.0526315789473684,
603
+ "grad_norm": 18.346025466918945,
604
+ "learning_rate": 5.2729044834307995e-06,
605
+ "loss": 4.5747,
606
+ "step": 600
607
+ },
608
+ {
609
+ "epoch": 1.0526315789473684,
610
+ "eval_q2q_data_loss": 4.405951499938965,
611
+ "eval_q2q_data_runtime": 5.5358,
612
+ "eval_q2q_data_samples_per_second": 317.749,
613
+ "eval_q2q_data_steps_per_second": 19.871,
614
+ "step": 600
615
+ },
616
+ {
617
+ "epoch": 1.0526315789473684,
618
+ "eval_q2p_data_loss": 4.791478633880615,
619
+ "eval_q2p_data_runtime": 7.389,
620
+ "eval_q2p_data_samples_per_second": 54.947,
621
+ "eval_q2p_data_steps_per_second": 3.519,
622
+ "step": 600
623
+ },
624
+ {
625
+ "epoch": 1.0701754385964912,
626
+ "grad_norm": 8.272171020507812,
627
+ "learning_rate": 5.175438596491229e-06,
628
+ "loss": 4.5296,
629
+ "step": 610
630
+ },
631
+ {
632
+ "epoch": 1.087719298245614,
633
+ "grad_norm": 8.837151527404785,
634
+ "learning_rate": 5.077972709551658e-06,
635
+ "loss": 4.4262,
636
+ "step": 620
637
+ },
638
+ {
639
+ "epoch": 1.1052631578947367,
640
+ "grad_norm": 13.43027400970459,
641
+ "learning_rate": 4.980506822612086e-06,
642
+ "loss": 4.5415,
643
+ "step": 630
644
+ },
645
+ {
646
+ "epoch": 1.1228070175438596,
647
+ "grad_norm": 8.466143608093262,
648
+ "learning_rate": 4.883040935672515e-06,
649
+ "loss": 4.5386,
650
+ "step": 640
651
+ },
652
+ {
653
+ "epoch": 1.1403508771929824,
654
+ "grad_norm": 10.755400657653809,
655
+ "learning_rate": 4.785575048732944e-06,
656
+ "loss": 4.4552,
657
+ "step": 650
658
+ },
659
+ {
660
+ "epoch": 1.1403508771929824,
661
+ "eval_q2q_data_loss": 4.363187789916992,
662
+ "eval_q2q_data_runtime": 5.5237,
663
+ "eval_q2q_data_samples_per_second": 318.449,
664
+ "eval_q2q_data_steps_per_second": 19.914,
665
+ "step": 650
666
+ },
667
+ {
668
+ "epoch": 1.1403508771929824,
669
+ "eval_q2p_data_loss": 4.810464382171631,
670
+ "eval_q2p_data_runtime": 7.469,
671
+ "eval_q2p_data_samples_per_second": 54.358,
672
+ "eval_q2p_data_steps_per_second": 3.481,
673
+ "step": 650
674
+ },
675
+ {
676
+ "epoch": 1.1578947368421053,
677
+ "grad_norm": 8.030132293701172,
678
+ "learning_rate": 4.688109161793373e-06,
679
+ "loss": 4.4473,
680
+ "step": 660
681
+ },
682
+ {
683
+ "epoch": 1.1754385964912282,
684
+ "grad_norm": 8.19764518737793,
685
+ "learning_rate": 4.590643274853801e-06,
686
+ "loss": 4.5069,
687
+ "step": 670
688
+ },
689
+ {
690
+ "epoch": 1.1929824561403508,
691
+ "grad_norm": 11.119821548461914,
692
+ "learning_rate": 4.4931773879142305e-06,
693
+ "loss": 4.5129,
694
+ "step": 680
695
+ },
696
+ {
697
+ "epoch": 1.2105263157894737,
698
+ "grad_norm": 9.186931610107422,
699
+ "learning_rate": 4.3957115009746595e-06,
700
+ "loss": 4.4611,
701
+ "step": 690
702
+ },
703
+ {
704
+ "epoch": 1.2280701754385965,
705
+ "grad_norm": 7.6313042640686035,
706
+ "learning_rate": 4.298245614035088e-06,
707
+ "loss": 4.5104,
708
+ "step": 700
709
+ },
710
+ {
711
+ "epoch": 1.2280701754385965,
712
+ "eval_q2q_data_loss": 4.353029727935791,
713
+ "eval_q2q_data_runtime": 5.559,
714
+ "eval_q2q_data_samples_per_second": 316.425,
715
+ "eval_q2q_data_steps_per_second": 19.788,
716
+ "step": 700
717
+ },
718
+ {
719
+ "epoch": 1.2280701754385965,
720
+ "eval_q2p_data_loss": 4.787461757659912,
721
+ "eval_q2p_data_runtime": 7.5053,
722
+ "eval_q2p_data_samples_per_second": 54.095,
723
+ "eval_q2p_data_steps_per_second": 3.464,
724
+ "step": 700
725
+ },
726
+ {
727
+ "epoch": 1.2456140350877192,
728
+ "grad_norm": 12.636022567749023,
729
+ "learning_rate": 4.200779727095517e-06,
730
+ "loss": 4.4742,
731
+ "step": 710
732
+ },
733
+ {
734
+ "epoch": 1.263157894736842,
735
+ "grad_norm": 16.598079681396484,
736
+ "learning_rate": 4.103313840155946e-06,
737
+ "loss": 4.4887,
738
+ "step": 720
739
+ },
740
+ {
741
+ "epoch": 1.280701754385965,
742
+ "grad_norm": 6.5720038414001465,
743
+ "learning_rate": 4.005847953216375e-06,
744
+ "loss": 4.406,
745
+ "step": 730
746
+ },
747
+ {
748
+ "epoch": 1.2982456140350878,
749
+ "grad_norm": 10.550318717956543,
750
+ "learning_rate": 3.908382066276803e-06,
751
+ "loss": 4.4049,
752
+ "step": 740
753
+ },
754
+ {
755
+ "epoch": 1.3157894736842106,
756
+ "grad_norm": 16.054428100585938,
757
+ "learning_rate": 3.8109161793372323e-06,
758
+ "loss": 4.4165,
759
+ "step": 750
760
+ },
761
+ {
762
+ "epoch": 1.3157894736842106,
763
+ "eval_q2q_data_loss": 4.348443031311035,
764
+ "eval_q2q_data_runtime": 5.5669,
765
+ "eval_q2q_data_samples_per_second": 315.976,
766
+ "eval_q2q_data_steps_per_second": 19.76,
767
+ "step": 750
768
+ },
769
+ {
770
+ "epoch": 1.3157894736842106,
771
+ "eval_q2p_data_loss": 4.786614894866943,
772
+ "eval_q2p_data_runtime": 7.508,
773
+ "eval_q2p_data_samples_per_second": 54.076,
774
+ "eval_q2p_data_steps_per_second": 3.463,
775
+ "step": 750
776
+ },
777
+ {
778
+ "epoch": 1.3333333333333333,
779
+ "grad_norm": 10.04055404663086,
780
+ "learning_rate": 3.713450292397661e-06,
781
+ "loss": 4.4274,
782
+ "step": 760
783
+ },
784
+ {
785
+ "epoch": 1.3508771929824561,
786
+ "grad_norm": 12.780068397521973,
787
+ "learning_rate": 3.61598440545809e-06,
788
+ "loss": 4.4855,
789
+ "step": 770
790
+ },
791
+ {
792
+ "epoch": 1.368421052631579,
793
+ "grad_norm": 10.54061222076416,
794
+ "learning_rate": 3.5185185185185187e-06,
795
+ "loss": 4.4571,
796
+ "step": 780
797
+ },
798
+ {
799
+ "epoch": 1.3859649122807016,
800
+ "grad_norm": 5.75900936126709,
801
+ "learning_rate": 3.421052631578948e-06,
802
+ "loss": 4.4307,
803
+ "step": 790
804
+ },
805
+ {
806
+ "epoch": 1.4035087719298245,
807
+ "grad_norm": 10.625808715820312,
808
+ "learning_rate": 3.3235867446393765e-06,
809
+ "loss": 4.4387,
810
+ "step": 800
811
+ },
812
+ {
813
+ "epoch": 1.4035087719298245,
814
+ "eval_q2q_data_loss": 4.345006465911865,
815
+ "eval_q2q_data_runtime": 5.5698,
816
+ "eval_q2q_data_samples_per_second": 315.808,
817
+ "eval_q2q_data_steps_per_second": 19.749,
818
+ "step": 800
819
+ },
820
+ {
821
+ "epoch": 1.4035087719298245,
822
+ "eval_q2p_data_loss": 4.762818813323975,
823
+ "eval_q2p_data_runtime": 7.5368,
824
+ "eval_q2p_data_samples_per_second": 53.869,
825
+ "eval_q2p_data_steps_per_second": 3.45,
826
+ "step": 800
827
+ },
828
+ {
829
+ "epoch": 1.4210526315789473,
830
+ "grad_norm": 9.662367820739746,
831
+ "learning_rate": 3.2261208576998056e-06,
832
+ "loss": 4.4592,
833
+ "step": 810
834
+ },
835
+ {
836
+ "epoch": 1.4385964912280702,
837
+ "grad_norm": 14.999639511108398,
838
+ "learning_rate": 3.1286549707602342e-06,
839
+ "loss": 4.4368,
840
+ "step": 820
841
+ },
842
+ {
843
+ "epoch": 1.456140350877193,
844
+ "grad_norm": 17.007898330688477,
845
+ "learning_rate": 3.0311890838206633e-06,
846
+ "loss": 4.4863,
847
+ "step": 830
848
+ },
849
+ {
850
+ "epoch": 1.4736842105263157,
851
+ "grad_norm": 14.116398811340332,
852
+ "learning_rate": 2.933723196881092e-06,
853
+ "loss": 4.463,
854
+ "step": 840
855
+ },
856
+ {
857
+ "epoch": 1.4912280701754386,
858
+ "grad_norm": 5.4955315589904785,
859
+ "learning_rate": 2.8362573099415206e-06,
860
+ "loss": 4.4113,
861
+ "step": 850
862
+ },
863
+ {
864
+ "epoch": 1.4912280701754386,
865
+ "eval_q2q_data_loss": 4.325167655944824,
866
+ "eval_q2q_data_runtime": 5.5814,
867
+ "eval_q2q_data_samples_per_second": 315.156,
868
+ "eval_q2q_data_steps_per_second": 19.708,
869
+ "step": 850
870
+ },
871
+ {
872
+ "epoch": 1.4912280701754386,
873
+ "eval_q2p_data_loss": 4.761044979095459,
874
+ "eval_q2p_data_runtime": 7.4985,
875
+ "eval_q2p_data_samples_per_second": 54.144,
876
+ "eval_q2p_data_steps_per_second": 3.467,
877
+ "step": 850
878
+ },
879
+ {
880
+ "epoch": 1.5087719298245614,
881
+ "grad_norm": 13.653097152709961,
882
+ "learning_rate": 2.7387914230019497e-06,
883
+ "loss": 4.4368,
884
+ "step": 860
885
+ },
886
+ {
887
+ "epoch": 1.526315789473684,
888
+ "grad_norm": 13.720170974731445,
889
+ "learning_rate": 2.6413255360623784e-06,
890
+ "loss": 4.4738,
891
+ "step": 870
892
+ },
893
+ {
894
+ "epoch": 1.543859649122807,
895
+ "grad_norm": 15.261076927185059,
896
+ "learning_rate": 2.5438596491228075e-06,
897
+ "loss": 4.4195,
898
+ "step": 880
899
+ },
900
+ {
901
+ "epoch": 1.5614035087719298,
902
+ "grad_norm": 10.974407196044922,
903
+ "learning_rate": 2.446393762183236e-06,
904
+ "loss": 4.4478,
905
+ "step": 890
906
+ },
907
+ {
908
+ "epoch": 1.5789473684210527,
909
+ "grad_norm": 10.83484935760498,
910
+ "learning_rate": 2.3489278752436648e-06,
911
+ "loss": 4.3849,
912
+ "step": 900
913
+ },
914
+ {
915
+ "epoch": 1.5789473684210527,
916
+ "eval_q2q_data_loss": 4.314022064208984,
917
+ "eval_q2q_data_runtime": 5.5727,
918
+ "eval_q2q_data_samples_per_second": 315.646,
919
+ "eval_q2q_data_steps_per_second": 19.739,
920
+ "step": 900
921
+ },
922
+ {
923
+ "epoch": 1.5789473684210527,
924
+ "eval_q2p_data_loss": 4.751864910125732,
925
+ "eval_q2p_data_runtime": 7.4934,
926
+ "eval_q2p_data_samples_per_second": 54.181,
927
+ "eval_q2p_data_steps_per_second": 3.47,
928
+ "step": 900
929
+ },
930
+ {
931
+ "epoch": 1.5964912280701755,
932
+ "grad_norm": 21.77918243408203,
933
+ "learning_rate": 2.2514619883040934e-06,
934
+ "loss": 4.4896,
935
+ "step": 910
936
+ },
937
+ {
938
+ "epoch": 1.6140350877192984,
939
+ "grad_norm": 7.528986930847168,
940
+ "learning_rate": 2.1539961013645225e-06,
941
+ "loss": 4.4301,
942
+ "step": 920
943
+ },
944
+ {
945
+ "epoch": 1.631578947368421,
946
+ "grad_norm": 8.18942928314209,
947
+ "learning_rate": 2.056530214424951e-06,
948
+ "loss": 4.4142,
949
+ "step": 930
950
+ },
951
+ {
952
+ "epoch": 1.6491228070175439,
953
+ "grad_norm": 10.001923561096191,
954
+ "learning_rate": 1.9590643274853803e-06,
955
+ "loss": 4.4582,
956
+ "step": 940
957
+ },
958
+ {
959
+ "epoch": 1.6666666666666665,
960
+ "grad_norm": 10.730441093444824,
961
+ "learning_rate": 1.861598440545809e-06,
962
+ "loss": 4.5075,
963
+ "step": 950
964
+ },
965
+ {
966
+ "epoch": 1.6666666666666665,
967
+ "eval_q2q_data_loss": 4.3189191818237305,
968
+ "eval_q2q_data_runtime": 5.5874,
969
+ "eval_q2q_data_samples_per_second": 314.816,
970
+ "eval_q2q_data_steps_per_second": 19.687,
971
+ "step": 950
972
+ },
973
+ {
974
+ "epoch": 1.6666666666666665,
975
+ "eval_q2p_data_loss": 4.725940704345703,
976
+ "eval_q2p_data_runtime": 7.514,
977
+ "eval_q2p_data_samples_per_second": 54.033,
978
+ "eval_q2p_data_steps_per_second": 3.46,
979
+ "step": 950
980
+ },
981
+ {
982
+ "epoch": 1.6842105263157894,
983
+ "grad_norm": 9.174509048461914,
984
+ "learning_rate": 1.7641325536062378e-06,
985
+ "loss": 4.4454,
986
+ "step": 960
987
+ },
988
+ {
989
+ "epoch": 1.7017543859649122,
990
+ "grad_norm": 11.805915832519531,
991
+ "learning_rate": 1.6666666666666667e-06,
992
+ "loss": 4.3547,
993
+ "step": 970
994
+ },
995
+ {
996
+ "epoch": 1.719298245614035,
997
+ "grad_norm": 9.230790138244629,
998
+ "learning_rate": 1.5692007797270955e-06,
999
+ "loss": 4.4016,
1000
+ "step": 980
1001
+ },
1002
+ {
1003
+ "epoch": 1.736842105263158,
1004
+ "grad_norm": 13.292176246643066,
1005
+ "learning_rate": 1.4717348927875244e-06,
1006
+ "loss": 4.4064,
1007
+ "step": 990
1008
+ },
1009
+ {
1010
+ "epoch": 1.7543859649122808,
1011
+ "grad_norm": 9.294161796569824,
1012
+ "learning_rate": 1.3742690058479533e-06,
1013
+ "loss": 4.4356,
1014
+ "step": 1000
1015
+ },
1016
+ {
1017
+ "epoch": 1.7543859649122808,
1018
+ "eval_q2q_data_loss": 4.3151326179504395,
1019
+ "eval_q2q_data_runtime": 5.5534,
1020
+ "eval_q2q_data_samples_per_second": 316.742,
1021
+ "eval_q2q_data_steps_per_second": 19.808,
1022
+ "step": 1000
1023
+ },
1024
+ {
1025
+ "epoch": 1.7543859649122808,
1026
+ "eval_q2p_data_loss": 4.727615833282471,
1027
+ "eval_q2p_data_runtime": 7.5335,
1028
+ "eval_q2p_data_samples_per_second": 53.893,
1029
+ "eval_q2p_data_steps_per_second": 3.451,
1030
+ "step": 1000
1031
+ },
1032
+ {
1033
+ "epoch": 1.7719298245614035,
1034
+ "grad_norm": 12.539956092834473,
1035
+ "learning_rate": 1.2768031189083821e-06,
1036
+ "loss": 4.4105,
1037
+ "step": 1010
1038
+ },
1039
+ {
1040
+ "epoch": 1.7894736842105263,
1041
+ "grad_norm": 15.329697608947754,
1042
+ "learning_rate": 1.179337231968811e-06,
1043
+ "loss": 4.4067,
1044
+ "step": 1020
1045
+ },
1046
+ {
1047
+ "epoch": 1.807017543859649,
1048
+ "grad_norm": 7.712077617645264,
1049
+ "learning_rate": 1.0818713450292399e-06,
1050
+ "loss": 4.4296,
1051
+ "step": 1030
1052
+ },
1053
+ {
1054
+ "epoch": 1.8245614035087718,
1055
+ "grad_norm": 7.909111976623535,
1056
+ "learning_rate": 9.844054580896685e-07,
1057
+ "loss": 4.4147,
1058
+ "step": 1040
1059
+ },
1060
+ {
1061
+ "epoch": 1.8421052631578947,
1062
+ "grad_norm": 16.474355697631836,
1063
+ "learning_rate": 8.869395711500975e-07,
1064
+ "loss": 4.3743,
1065
+ "step": 1050
1066
+ },
1067
+ {
1068
+ "epoch": 1.8421052631578947,
1069
+ "eval_q2q_data_loss": 4.313626289367676,
1070
+ "eval_q2q_data_runtime": 5.5976,
1071
+ "eval_q2q_data_samples_per_second": 314.244,
1072
+ "eval_q2q_data_steps_per_second": 19.651,
1073
+ "step": 1050
1074
+ },
1075
+ {
1076
+ "epoch": 1.8421052631578947,
1077
+ "eval_q2p_data_loss": 4.7181901931762695,
1078
+ "eval_q2p_data_runtime": 7.5322,
1079
+ "eval_q2p_data_samples_per_second": 53.902,
1080
+ "eval_q2p_data_steps_per_second": 3.452,
1081
+ "step": 1050
1082
+ },
1083
+ {
1084
+ "epoch": 1.8596491228070176,
1085
+ "grad_norm": 11.424127578735352,
1086
+ "learning_rate": 7.894736842105263e-07,
1087
+ "loss": 4.4065,
1088
+ "step": 1060
1089
+ },
1090
+ {
1091
+ "epoch": 1.8771929824561404,
1092
+ "grad_norm": 8.293243408203125,
1093
+ "learning_rate": 6.920077972709552e-07,
1094
+ "loss": 4.4025,
1095
+ "step": 1070
1096
+ },
1097
+ {
1098
+ "epoch": 1.8947368421052633,
1099
+ "grad_norm": 11.082077026367188,
1100
+ "learning_rate": 5.94541910331384e-07,
1101
+ "loss": 4.3912,
1102
+ "step": 1080
1103
+ },
1104
+ {
1105
+ "epoch": 1.912280701754386,
1106
+ "grad_norm": 13.221600532531738,
1107
+ "learning_rate": 4.970760233918129e-07,
1108
+ "loss": 4.3731,
1109
+ "step": 1090
1110
+ },
1111
+ {
1112
+ "epoch": 1.9298245614035088,
1113
+ "grad_norm": 16.041154861450195,
1114
+ "learning_rate": 3.996101364522417e-07,
1115
+ "loss": 4.3817,
1116
+ "step": 1100
1117
+ },
1118
+ {
1119
+ "epoch": 1.9298245614035088,
1120
+ "eval_q2q_data_loss": 4.311989784240723,
1121
+ "eval_q2q_data_runtime": 5.5712,
1122
+ "eval_q2q_data_samples_per_second": 315.734,
1123
+ "eval_q2q_data_steps_per_second": 19.745,
1124
+ "step": 1100
1125
+ },
1126
+ {
1127
+ "epoch": 1.9298245614035088,
1128
+ "eval_q2p_data_loss": 4.735711097717285,
1129
+ "eval_q2p_data_runtime": 7.4899,
1130
+ "eval_q2p_data_samples_per_second": 54.207,
1131
+ "eval_q2p_data_steps_per_second": 3.471,
1132
+ "step": 1100
1133
+ },
1134
+ {
1135
+ "epoch": 1.9473684210526314,
1136
+ "grad_norm": 8.501266479492188,
1137
+ "learning_rate": 3.021442495126706e-07,
1138
+ "loss": 4.4305,
1139
+ "step": 1110
1140
+ },
1141
+ {
1142
+ "epoch": 1.9649122807017543,
1143
+ "grad_norm": 7.625467300415039,
1144
+ "learning_rate": 2.046783625730994e-07,
1145
+ "loss": 4.3914,
1146
+ "step": 1120
1147
+ },
1148
+ {
1149
+ "epoch": 1.9824561403508771,
1150
+ "grad_norm": 11.992876052856445,
1151
+ "learning_rate": 1.0721247563352827e-07,
1152
+ "loss": 4.4753,
1153
+ "step": 1130
1154
+ },
1155
+ {
1156
+ "epoch": 2.0,
1157
+ "grad_norm": 5.4963908195495605,
1158
+ "learning_rate": 9.746588693957116e-09,
1159
+ "loss": 4.4536,
1160
+ "step": 1140
1161
+ }
1162
+ ],
1163
+ "logging_steps": 10,
1164
+ "max_steps": 1140,
1165
+ "num_input_tokens_seen": 0,
1166
+ "num_train_epochs": 2,
1167
+ "save_steps": 200,
1168
+ "stateful_callbacks": {
1169
+ "TrainerControl": {
1170
+ "args": {
1171
+ "should_epoch_stop": false,
1172
+ "should_evaluate": false,
1173
+ "should_log": false,
1174
+ "should_save": true,
1175
+ "should_training_stop": true
1176
+ },
1177
+ "attributes": {}
1178
+ }
1179
+ },
1180
+ "total_flos": 0.0,
1181
+ "train_batch_size": 36,
1182
+ "trial_name": null,
1183
+ "trial_params": null
1184
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f02d416f17419a4600b5130e463c5c4c20b0ba7d76e1597f3d0f857d0d83a2eb
3
+ size 5624