Initial release of Meitei-RoBERTa-Base (Bengali Script) with training logs and model card.
Browse files- README.md +62 -0
- checkpoint-1380/config.json +26 -0
- checkpoint-1380/model.safetensors +3 -0
- checkpoint-1380/optimizer.pt +3 -0
- checkpoint-1380/rng_state.pth +3 -0
- checkpoint-1380/scaler.pt +3 -0
- checkpoint-1380/scheduler.pt +3 -0
- checkpoint-1380/special_tokens_map.json +37 -0
- checkpoint-1380/tokenizer.json +0 -0
- checkpoint-1380/tokenizer_config.json +53 -0
- checkpoint-1380/trainer_state.json +48 -0
- checkpoint-1380/training_args.bin +3 -0
- checkpoint-2760/config.json +26 -0
- checkpoint-2760/model.safetensors +3 -0
- checkpoint-2760/optimizer.pt +3 -0
- checkpoint-2760/rng_state.pth +3 -0
- checkpoint-2760/scaler.pt +3 -0
- checkpoint-2760/scheduler.pt +3 -0
- checkpoint-2760/special_tokens_map.json +37 -0
- checkpoint-2760/tokenizer.json +0 -0
- checkpoint-2760/tokenizer_config.json +53 -0
- checkpoint-2760/trainer_state.json +69 -0
- checkpoint-2760/training_args.bin +3 -0
- checkpoint-4140/config.json +26 -0
- checkpoint-4140/model.safetensors +3 -0
- checkpoint-4140/optimizer.pt +3 -0
- checkpoint-4140/rng_state.pth +3 -0
- checkpoint-4140/scaler.pt +3 -0
- checkpoint-4140/scheduler.pt +3 -0
- checkpoint-4140/special_tokens_map.json +37 -0
- checkpoint-4140/tokenizer.json +0 -0
- checkpoint-4140/tokenizer_config.json +53 -0
- checkpoint-4140/trainer_state.json +90 -0
- checkpoint-4140/training_args.bin +3 -0
- config.json +26 -0
- model.safetensors +3 -0
- special_tokens_map.json +37 -0
- tokenizer.json +0 -0
- tokenizer_config.json +53 -0
- training_args.bin +3 -0
- training_args.json +148 -0
- training_loss_history.csv +9 -0
README.md
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
---
|
| 3 |
+
model_id: Badnyal/Meitei-RoBERTa
|
| 4 |
+
language: "mni" # ISO 639-2 code for Manipuri (Meitei)
|
| 5 |
+
tags:
|
| 6 |
+
- masked-language-modeling
|
| 7 |
+
- transformer
|
| 8 |
+
- roberta
|
| 9 |
+
- meitei
|
| 10 |
+
- low-resource
|
| 11 |
+
- monolingual
|
| 12 |
+
license: "cc-by-sa-4.0"
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
# Meitei-RoBERTa-Base (Bengali Script)
|
| 16 |
+
|
| 17 |
+
This is a **monolingual RoBERTa Base model** pre-trained from scratch on the entire **Meitei Monolingual Corpus** (MWirelabs/meitei-monolingual-corpus).
|
| 18 |
+
It establishes the **State-of-the-Art foundational language model** for Meitei (Manipuri) in the Bengali script by leveraging a custom tokenizer and robust RoBERTa training objectives.
|
| 19 |
+
|
| 20 |
+
## 💾 Model Details
|
| 21 |
+
|
| 22 |
+
| Detail | Value |
|
| 23 |
+
| :--- | :--- |
|
| 24 |
+
| **Architecture** | RoBERTa Base (12 Layers, 12 Heads, 768 Hidden) |
|
| 25 |
+
| **Total Parameters** | 126,031,648 |
|
| 26 |
+
| **Pre-training Task** | Masked Language Modeling (MLM) |
|
| 27 |
+
| **Tokenizer** | Byte-Level BPE |
|
| 28 |
+
| **Vocabulary Size** | 52000 |
|
| 29 |
+
| **Context Length** | 512 |
|
| 30 |
+
|
| 31 |
+
## ⚙️ Training Information
|
| 32 |
+
|
| 33 |
+
| Parameter | Value |
|
| 34 |
+
| :--- | :--- |
|
| 35 |
+
| **Dataset** | MWirelabs/meitei-monolingual-corpus (train split) |
|
| 36 |
+
| **Data Size (Chunks)** | 353,123 blocks of 512 tokens |
|
| 37 |
+
| **Effective Batch Size** | 256 |
|
| 38 |
+
| **Learning Rate** | 0.0006 |
|
| 39 |
+
| **Epochs** | 3 |
|
| 40 |
+
| **Final Loss** | 4.18550 |
|
| 41 |
+
|
| 42 |
+
## 📈 Loss Curve (For Visual Chart)
|
| 43 |
+
|
| 44 |
+
The `training_loss_history.csv` file, included in this repository, contains the sequential loss data over training steps, allowing users to plot the convergence curve.
|
| 45 |
+
|
| 46 |
+
## 💡 How to Use
|
| 47 |
+
|
| 48 |
+
```python
|
| 49 |
+
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
| 50 |
+
|
| 51 |
+
# The repository ID is used to load both model and tokenizer
|
| 52 |
+
REPO_ID = "Badnyal/Meitei-RoBERTa"
|
| 53 |
+
|
| 54 |
+
tokenizer = AutoTokenizer.from_pretrained(REPO_ID)
|
| 55 |
+
model = AutoModelForMaskedLM.from_pretrained(REPO_ID)
|
| 56 |
+
|
| 57 |
+
# Example: Tokenizing a Meitei sentence
|
| 58 |
+
text = "ꯃꯤꯑꯣꯏꯕꯥ ꯑꯃꯥ ꯊꯤꯡꯖꯤꯟꯕꯥ ꯇꯥꯕꯥ꯫ ꯈꯣꯡꯎꯞ ꯄꯤꯕꯥ ꯇꯥꯕꯥ꯫"
|
| 59 |
+
inputs = tokenizer(text, return_tensors="pt")
|
| 60 |
+
# The model can now be used for feature extraction, zero-shot inference,
|
| 61 |
+
# or fine-tuned for downstream tasks like Named Entity Recognition (NER) or classification.
|
| 62 |
+
```
|
checkpoint-1380/config.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"RobertaForMaskedLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"classifier_dropout": null,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 1,
|
| 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": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"pad_token_id": 3,
|
| 21 |
+
"position_embedding_type": "absolute",
|
| 22 |
+
"transformers_version": "4.57.1",
|
| 23 |
+
"type_vocab_size": 1,
|
| 24 |
+
"use_cache": true,
|
| 25 |
+
"vocab_size": 52000
|
| 26 |
+
}
|
checkpoint-1380/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8af0f5869472238d4377c634b3ffce4302a2ccfbc3b69c79f0810e7cbb895e0d
|
| 3 |
+
size 504150808
|
checkpoint-1380/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a60354a24d1df0c6df3b4a8e446ff4f31d51bd0b46f63089079b6aef2b8eeb89
|
| 3 |
+
size 1008422603
|
checkpoint-1380/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c4bdb8c0daa13c0e6fd1faeb8d0cd13cf72d338a3572436eab98a1683ff226cc
|
| 3 |
+
size 14645
|
checkpoint-1380/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:20f288acd21e6afcbe6ed0dbfc211ea448b3d5dc72c8727ed48b0d623bdb674c
|
| 3 |
+
size 1383
|
checkpoint-1380/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cd8947e9bce174e8553636e8f6971f077f77a936b51b52cdd7070a5104f944ce
|
| 3 |
+
size 1465
|
checkpoint-1380/special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "</s>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"mask_token": {
|
| 17 |
+
"content": "<mask>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"pad_token": {
|
| 24 |
+
"content": "<pad>",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "<unk>",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
checkpoint-1380/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-1380/tokenizer_config.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "</s>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"2": {
|
| 20 |
+
"content": "<unk>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"3": {
|
| 28 |
+
"content": "<pad>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"4": {
|
| 36 |
+
"content": "<mask>",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"bos_token": "<s>",
|
| 45 |
+
"clean_up_tokenization_spaces": false,
|
| 46 |
+
"eos_token": "</s>",
|
| 47 |
+
"extra_special_tokens": {},
|
| 48 |
+
"mask_token": "<mask>",
|
| 49 |
+
"model_max_length": 512,
|
| 50 |
+
"pad_token": "<pad>",
|
| 51 |
+
"tokenizer_class": "PreTrainedTokenizerFast",
|
| 52 |
+
"unk_token": "<unk>"
|
| 53 |
+
}
|
checkpoint-1380/trainer_state.json
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 1.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 1380,
|
| 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.3624501631025734,
|
| 14 |
+
"grad_norm": 0.4742291271686554,
|
| 15 |
+
"learning_rate": 2.9939999999999998e-05,
|
| 16 |
+
"loss": 6.4565,
|
| 17 |
+
"step": 500
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.7249003262051468,
|
| 21 |
+
"grad_norm": 0.5359567403793335,
|
| 22 |
+
"learning_rate": 5.994e-05,
|
| 23 |
+
"loss": 4.3201,
|
| 24 |
+
"step": 1000
|
| 25 |
+
}
|
| 26 |
+
],
|
| 27 |
+
"logging_steps": 500,
|
| 28 |
+
"max_steps": 4140,
|
| 29 |
+
"num_input_tokens_seen": 0,
|
| 30 |
+
"num_train_epochs": 3,
|
| 31 |
+
"save_steps": 500,
|
| 32 |
+
"stateful_callbacks": {
|
| 33 |
+
"TrainerControl": {
|
| 34 |
+
"args": {
|
| 35 |
+
"should_epoch_stop": false,
|
| 36 |
+
"should_evaluate": false,
|
| 37 |
+
"should_log": false,
|
| 38 |
+
"should_save": true,
|
| 39 |
+
"should_training_stop": false
|
| 40 |
+
},
|
| 41 |
+
"attributes": {}
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"total_flos": 9.296697231281357e+16,
|
| 45 |
+
"train_batch_size": 32,
|
| 46 |
+
"trial_name": null,
|
| 47 |
+
"trial_params": null
|
| 48 |
+
}
|
checkpoint-1380/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f7549eb3763a9c4d7cf3ff0edbfd336c1ffe1406f78df5cf66043df3cc176d3
|
| 3 |
+
size 5777
|
checkpoint-2760/config.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"RobertaForMaskedLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"classifier_dropout": null,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 1,
|
| 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": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"pad_token_id": 3,
|
| 21 |
+
"position_embedding_type": "absolute",
|
| 22 |
+
"transformers_version": "4.57.1",
|
| 23 |
+
"type_vocab_size": 1,
|
| 24 |
+
"use_cache": true,
|
| 25 |
+
"vocab_size": 52000
|
| 26 |
+
}
|
checkpoint-2760/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bed97d14a18f7ace7e9aec9afefdefa85534f77d841b11ca5ae070eb5110ec7e
|
| 3 |
+
size 504150808
|
checkpoint-2760/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e13e016912ff159d50148a09ee6c18281d6259cb98424ce6f62dc0f17c53eaf9
|
| 3 |
+
size 1008422603
|
checkpoint-2760/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d9e4077dba68b8bcdeb028955aafe1db7a8f256e51e622a2105cc82971423ff7
|
| 3 |
+
size 14645
|
checkpoint-2760/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ec897a068d90bd14cd2b4ebbf344be41c3f69d39b2d7eb870c189a7308eaf18
|
| 3 |
+
size 1383
|
checkpoint-2760/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:08586d77a1761a3a1ec1920d4c0c81326d13398b5c337c4a0c66b11781ed38f8
|
| 3 |
+
size 1465
|
checkpoint-2760/special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "</s>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"mask_token": {
|
| 17 |
+
"content": "<mask>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"pad_token": {
|
| 24 |
+
"content": "<pad>",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "<unk>",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
checkpoint-2760/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-2760/tokenizer_config.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "</s>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"2": {
|
| 20 |
+
"content": "<unk>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"3": {
|
| 28 |
+
"content": "<pad>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"4": {
|
| 36 |
+
"content": "<mask>",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"bos_token": "<s>",
|
| 45 |
+
"clean_up_tokenization_spaces": false,
|
| 46 |
+
"eos_token": "</s>",
|
| 47 |
+
"extra_special_tokens": {},
|
| 48 |
+
"mask_token": "<mask>",
|
| 49 |
+
"model_max_length": 512,
|
| 50 |
+
"pad_token": "<pad>",
|
| 51 |
+
"tokenizer_class": "PreTrainedTokenizerFast",
|
| 52 |
+
"unk_token": "<unk>"
|
| 53 |
+
}
|
checkpoint-2760/trainer_state.json
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 2.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 2760,
|
| 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.3624501631025734,
|
| 14 |
+
"grad_norm": 0.4742291271686554,
|
| 15 |
+
"learning_rate": 2.9939999999999998e-05,
|
| 16 |
+
"loss": 6.4565,
|
| 17 |
+
"step": 500
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.7249003262051468,
|
| 21 |
+
"grad_norm": 0.5359567403793335,
|
| 22 |
+
"learning_rate": 5.994e-05,
|
| 23 |
+
"loss": 4.3201,
|
| 24 |
+
"step": 1000
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 1.0869880391446176,
|
| 28 |
+
"grad_norm": 0.4428270161151886,
|
| 29 |
+
"learning_rate": 8.994e-05,
|
| 30 |
+
"loss": 4.2592,
|
| 31 |
+
"step": 1500
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 1.449438202247191,
|
| 35 |
+
"grad_norm": 0.4446365237236023,
|
| 36 |
+
"learning_rate": 0.00011993999999999998,
|
| 37 |
+
"loss": 4.2331,
|
| 38 |
+
"step": 2000
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 1.8118883653497644,
|
| 42 |
+
"grad_norm": 0.7092395424842834,
|
| 43 |
+
"learning_rate": 0.00014994,
|
| 44 |
+
"loss": 4.2168,
|
| 45 |
+
"step": 2500
|
| 46 |
+
}
|
| 47 |
+
],
|
| 48 |
+
"logging_steps": 500,
|
| 49 |
+
"max_steps": 4140,
|
| 50 |
+
"num_input_tokens_seen": 0,
|
| 51 |
+
"num_train_epochs": 3,
|
| 52 |
+
"save_steps": 500,
|
| 53 |
+
"stateful_callbacks": {
|
| 54 |
+
"TrainerControl": {
|
| 55 |
+
"args": {
|
| 56 |
+
"should_epoch_stop": false,
|
| 57 |
+
"should_evaluate": false,
|
| 58 |
+
"should_log": false,
|
| 59 |
+
"should_save": true,
|
| 60 |
+
"should_training_stop": false
|
| 61 |
+
},
|
| 62 |
+
"attributes": {}
|
| 63 |
+
}
|
| 64 |
+
},
|
| 65 |
+
"total_flos": 1.8593394462562714e+17,
|
| 66 |
+
"train_batch_size": 32,
|
| 67 |
+
"trial_name": null,
|
| 68 |
+
"trial_params": null
|
| 69 |
+
}
|
checkpoint-2760/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f7549eb3763a9c4d7cf3ff0edbfd336c1ffe1406f78df5cf66043df3cc176d3
|
| 3 |
+
size 5777
|
checkpoint-4140/config.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"RobertaForMaskedLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"classifier_dropout": null,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 1,
|
| 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": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"pad_token_id": 3,
|
| 21 |
+
"position_embedding_type": "absolute",
|
| 22 |
+
"transformers_version": "4.57.1",
|
| 23 |
+
"type_vocab_size": 1,
|
| 24 |
+
"use_cache": true,
|
| 25 |
+
"vocab_size": 52000
|
| 26 |
+
}
|
checkpoint-4140/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:34cf01a40bc5231ecaa9bfa4063af838520f8fe60cea00e5b89c65c22069fb18
|
| 3 |
+
size 504150808
|
checkpoint-4140/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f73c93edeec24497db533fd9ccbac7ecdf724d6cbca38c80ceca4956d02c0f0
|
| 3 |
+
size 1008422603
|
checkpoint-4140/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ed945a78bb0c2401d9f48d0b285919b4b5118bad5e82a7f1963cd763f67f710
|
| 3 |
+
size 14645
|
checkpoint-4140/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:37c28d86d87229869f85f25e0381969a9d63089996702db9efd08cbb3ec61591
|
| 3 |
+
size 1383
|
checkpoint-4140/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:14b8e841653051ccc7faf4c2062bb12f15b917ad76a00cb9241f9f0029123179
|
| 3 |
+
size 1465
|
checkpoint-4140/special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "</s>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"mask_token": {
|
| 17 |
+
"content": "<mask>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"pad_token": {
|
| 24 |
+
"content": "<pad>",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "<unk>",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
checkpoint-4140/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-4140/tokenizer_config.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "</s>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"2": {
|
| 20 |
+
"content": "<unk>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"3": {
|
| 28 |
+
"content": "<pad>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"4": {
|
| 36 |
+
"content": "<mask>",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"bos_token": "<s>",
|
| 45 |
+
"clean_up_tokenization_spaces": false,
|
| 46 |
+
"eos_token": "</s>",
|
| 47 |
+
"extra_special_tokens": {},
|
| 48 |
+
"mask_token": "<mask>",
|
| 49 |
+
"model_max_length": 512,
|
| 50 |
+
"pad_token": "<pad>",
|
| 51 |
+
"tokenizer_class": "PreTrainedTokenizerFast",
|
| 52 |
+
"unk_token": "<unk>"
|
| 53 |
+
}
|
checkpoint-4140/trainer_state.json
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 4140,
|
| 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.3624501631025734,
|
| 14 |
+
"grad_norm": 0.4742291271686554,
|
| 15 |
+
"learning_rate": 2.9939999999999998e-05,
|
| 16 |
+
"loss": 6.4565,
|
| 17 |
+
"step": 500
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.7249003262051468,
|
| 21 |
+
"grad_norm": 0.5359567403793335,
|
| 22 |
+
"learning_rate": 5.994e-05,
|
| 23 |
+
"loss": 4.3201,
|
| 24 |
+
"step": 1000
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 1.0869880391446176,
|
| 28 |
+
"grad_norm": 0.4428270161151886,
|
| 29 |
+
"learning_rate": 8.994e-05,
|
| 30 |
+
"loss": 4.2592,
|
| 31 |
+
"step": 1500
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 1.449438202247191,
|
| 35 |
+
"grad_norm": 0.4446365237236023,
|
| 36 |
+
"learning_rate": 0.00011993999999999998,
|
| 37 |
+
"loss": 4.2331,
|
| 38 |
+
"step": 2000
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 1.8118883653497644,
|
| 42 |
+
"grad_norm": 0.7092395424842834,
|
| 43 |
+
"learning_rate": 0.00014994,
|
| 44 |
+
"loss": 4.2168,
|
| 45 |
+
"step": 2500
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 2.1739760782892352,
|
| 49 |
+
"grad_norm": 0.3681005835533142,
|
| 50 |
+
"learning_rate": 0.00017994,
|
| 51 |
+
"loss": 4.2023,
|
| 52 |
+
"step": 3000
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 2.5364262413918084,
|
| 56 |
+
"grad_norm": 0.5603296756744385,
|
| 57 |
+
"learning_rate": 0.00020993999999999999,
|
| 58 |
+
"loss": 4.1945,
|
| 59 |
+
"step": 3500
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 2.898876404494382,
|
| 63 |
+
"grad_norm": 0.34806573390960693,
|
| 64 |
+
"learning_rate": 0.00023993999999999996,
|
| 65 |
+
"loss": 4.1855,
|
| 66 |
+
"step": 4000
|
| 67 |
+
}
|
| 68 |
+
],
|
| 69 |
+
"logging_steps": 500,
|
| 70 |
+
"max_steps": 4140,
|
| 71 |
+
"num_input_tokens_seen": 0,
|
| 72 |
+
"num_train_epochs": 3,
|
| 73 |
+
"save_steps": 500,
|
| 74 |
+
"stateful_callbacks": {
|
| 75 |
+
"TrainerControl": {
|
| 76 |
+
"args": {
|
| 77 |
+
"should_epoch_stop": false,
|
| 78 |
+
"should_evaluate": false,
|
| 79 |
+
"should_log": false,
|
| 80 |
+
"should_save": true,
|
| 81 |
+
"should_training_stop": true
|
| 82 |
+
},
|
| 83 |
+
"attributes": {}
|
| 84 |
+
}
|
| 85 |
+
},
|
| 86 |
+
"total_flos": 2.789009169384407e+17,
|
| 87 |
+
"train_batch_size": 32,
|
| 88 |
+
"trial_name": null,
|
| 89 |
+
"trial_params": null
|
| 90 |
+
}
|
checkpoint-4140/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f7549eb3763a9c4d7cf3ff0edbfd336c1ffe1406f78df5cf66043df3cc176d3
|
| 3 |
+
size 5777
|
config.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"RobertaForMaskedLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"classifier_dropout": null,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 1,
|
| 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": 514,
|
| 17 |
+
"model_type": "roberta",
|
| 18 |
+
"num_attention_heads": 12,
|
| 19 |
+
"num_hidden_layers": 12,
|
| 20 |
+
"pad_token_id": 3,
|
| 21 |
+
"position_embedding_type": "absolute",
|
| 22 |
+
"transformers_version": "4.57.1",
|
| 23 |
+
"type_vocab_size": 1,
|
| 24 |
+
"use_cache": true,
|
| 25 |
+
"vocab_size": 52000
|
| 26 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:34cf01a40bc5231ecaa9bfa4063af838520f8fe60cea00e5b89c65c22069fb18
|
| 3 |
+
size 504150808
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "</s>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"mask_token": {
|
| 17 |
+
"content": "<mask>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"pad_token": {
|
| 24 |
+
"content": "<pad>",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "<unk>",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "</s>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"2": {
|
| 20 |
+
"content": "<unk>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"3": {
|
| 28 |
+
"content": "<pad>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"4": {
|
| 36 |
+
"content": "<mask>",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"bos_token": "<s>",
|
| 45 |
+
"clean_up_tokenization_spaces": false,
|
| 46 |
+
"eos_token": "</s>",
|
| 47 |
+
"extra_special_tokens": {},
|
| 48 |
+
"mask_token": "<mask>",
|
| 49 |
+
"model_max_length": 512,
|
| 50 |
+
"pad_token": "<pad>",
|
| 51 |
+
"tokenizer_class": "PreTrainedTokenizerFast",
|
| 52 |
+
"unk_token": "<unk>"
|
| 53 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f7549eb3763a9c4d7cf3ff0edbfd336c1ffe1406f78df5cf66043df3cc176d3
|
| 3 |
+
size 5777
|
training_args.json
ADDED
|
@@ -0,0 +1,148 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"output_dir": "Meitei-RoBERTa-Base-MLM",
|
| 3 |
+
"overwrite_output_dir": true,
|
| 4 |
+
"do_train": false,
|
| 5 |
+
"do_eval": false,
|
| 6 |
+
"do_predict": false,
|
| 7 |
+
"eval_strategy": "no",
|
| 8 |
+
"prediction_loss_only": false,
|
| 9 |
+
"per_device_train_batch_size": 32,
|
| 10 |
+
"per_device_eval_batch_size": 8,
|
| 11 |
+
"per_gpu_train_batch_size": null,
|
| 12 |
+
"per_gpu_eval_batch_size": null,
|
| 13 |
+
"gradient_accumulation_steps": 8,
|
| 14 |
+
"eval_accumulation_steps": null,
|
| 15 |
+
"eval_delay": 0,
|
| 16 |
+
"torch_empty_cache_steps": null,
|
| 17 |
+
"learning_rate": 0.0006,
|
| 18 |
+
"weight_decay": 0.01,
|
| 19 |
+
"adam_beta1": 0.9,
|
| 20 |
+
"adam_beta2": 0.999,
|
| 21 |
+
"adam_epsilon": 1e-08,
|
| 22 |
+
"max_grad_norm": 1.0,
|
| 23 |
+
"num_train_epochs": 3,
|
| 24 |
+
"max_steps": -1,
|
| 25 |
+
"lr_scheduler_type": "linear",
|
| 26 |
+
"lr_scheduler_kwargs": {},
|
| 27 |
+
"warmup_ratio": 0.0,
|
| 28 |
+
"warmup_steps": 10000,
|
| 29 |
+
"log_level": "passive",
|
| 30 |
+
"log_level_replica": "warning",
|
| 31 |
+
"log_on_each_node": true,
|
| 32 |
+
"logging_dir": "./logs",
|
| 33 |
+
"logging_strategy": "steps",
|
| 34 |
+
"logging_first_step": false,
|
| 35 |
+
"logging_steps": 500,
|
| 36 |
+
"logging_nan_inf_filter": true,
|
| 37 |
+
"save_strategy": "epoch",
|
| 38 |
+
"save_steps": 500,
|
| 39 |
+
"save_total_limit": 3,
|
| 40 |
+
"save_safetensors": true,
|
| 41 |
+
"save_on_each_node": false,
|
| 42 |
+
"save_only_model": false,
|
| 43 |
+
"restore_callback_states_from_checkpoint": false,
|
| 44 |
+
"no_cuda": false,
|
| 45 |
+
"use_cpu": false,
|
| 46 |
+
"use_mps_device": false,
|
| 47 |
+
"seed": 42,
|
| 48 |
+
"data_seed": null,
|
| 49 |
+
"jit_mode_eval": false,
|
| 50 |
+
"bf16": false,
|
| 51 |
+
"fp16": true,
|
| 52 |
+
"fp16_opt_level": "O1",
|
| 53 |
+
"half_precision_backend": "auto",
|
| 54 |
+
"bf16_full_eval": false,
|
| 55 |
+
"fp16_full_eval": false,
|
| 56 |
+
"tf32": null,
|
| 57 |
+
"local_rank": 0,
|
| 58 |
+
"ddp_backend": null,
|
| 59 |
+
"tpu_num_cores": null,
|
| 60 |
+
"tpu_metrics_debug": false,
|
| 61 |
+
"debug": [],
|
| 62 |
+
"dataloader_drop_last": false,
|
| 63 |
+
"eval_steps": null,
|
| 64 |
+
"dataloader_num_workers": 0,
|
| 65 |
+
"dataloader_prefetch_factor": null,
|
| 66 |
+
"past_index": -1,
|
| 67 |
+
"run_name": null,
|
| 68 |
+
"disable_tqdm": false,
|
| 69 |
+
"remove_unused_columns": true,
|
| 70 |
+
"label_names": null,
|
| 71 |
+
"load_best_model_at_end": false,
|
| 72 |
+
"metric_for_best_model": null,
|
| 73 |
+
"greater_is_better": null,
|
| 74 |
+
"ignore_data_skip": false,
|
| 75 |
+
"fsdp": [],
|
| 76 |
+
"fsdp_min_num_params": 0,
|
| 77 |
+
"fsdp_config": {
|
| 78 |
+
"min_num_params": 0,
|
| 79 |
+
"xla": false,
|
| 80 |
+
"xla_fsdp_v2": false,
|
| 81 |
+
"xla_fsdp_grad_ckpt": false
|
| 82 |
+
},
|
| 83 |
+
"fsdp_transformer_layer_cls_to_wrap": null,
|
| 84 |
+
"accelerator_config": {
|
| 85 |
+
"split_batches": false,
|
| 86 |
+
"dispatch_batches": null,
|
| 87 |
+
"even_batches": true,
|
| 88 |
+
"use_seedable_sampler": true,
|
| 89 |
+
"non_blocking": false,
|
| 90 |
+
"gradient_accumulation_kwargs": null
|
| 91 |
+
},
|
| 92 |
+
"parallelism_config": null,
|
| 93 |
+
"deepspeed": null,
|
| 94 |
+
"label_smoothing_factor": 0.0,
|
| 95 |
+
"optim": "adamw_torch",
|
| 96 |
+
"optim_args": null,
|
| 97 |
+
"adafactor": false,
|
| 98 |
+
"group_by_length": false,
|
| 99 |
+
"length_column_name": "length",
|
| 100 |
+
"report_to": [
|
| 101 |
+
"tensorboard"
|
| 102 |
+
],
|
| 103 |
+
"project": "huggingface",
|
| 104 |
+
"trackio_space_id": "trackio",
|
| 105 |
+
"ddp_find_unused_parameters": null,
|
| 106 |
+
"ddp_bucket_cap_mb": null,
|
| 107 |
+
"ddp_broadcast_buffers": null,
|
| 108 |
+
"dataloader_pin_memory": true,
|
| 109 |
+
"dataloader_persistent_workers": false,
|
| 110 |
+
"skip_memory_metrics": true,
|
| 111 |
+
"use_legacy_prediction_loop": false,
|
| 112 |
+
"push_to_hub": false,
|
| 113 |
+
"resume_from_checkpoint": null,
|
| 114 |
+
"hub_model_id": null,
|
| 115 |
+
"hub_strategy": "every_save",
|
| 116 |
+
"hub_token": "<HUB_TOKEN>",
|
| 117 |
+
"hub_private_repo": null,
|
| 118 |
+
"hub_always_push": false,
|
| 119 |
+
"hub_revision": null,
|
| 120 |
+
"gradient_checkpointing": true,
|
| 121 |
+
"gradient_checkpointing_kwargs": null,
|
| 122 |
+
"include_inputs_for_metrics": false,
|
| 123 |
+
"include_for_metrics": [],
|
| 124 |
+
"eval_do_concat_batches": true,
|
| 125 |
+
"fp16_backend": "auto",
|
| 126 |
+
"push_to_hub_model_id": null,
|
| 127 |
+
"push_to_hub_organization": null,
|
| 128 |
+
"push_to_hub_token": "<PUSH_TO_HUB_TOKEN>",
|
| 129 |
+
"mp_parameters": "",
|
| 130 |
+
"auto_find_batch_size": false,
|
| 131 |
+
"full_determinism": false,
|
| 132 |
+
"torchdynamo": null,
|
| 133 |
+
"ray_scope": "last",
|
| 134 |
+
"ddp_timeout": 1800,
|
| 135 |
+
"torch_compile": false,
|
| 136 |
+
"torch_compile_backend": null,
|
| 137 |
+
"torch_compile_mode": null,
|
| 138 |
+
"include_tokens_per_second": false,
|
| 139 |
+
"include_num_input_tokens_seen": "no",
|
| 140 |
+
"neftune_noise_alpha": null,
|
| 141 |
+
"optim_target_modules": null,
|
| 142 |
+
"batch_eval_metrics": false,
|
| 143 |
+
"eval_on_start": false,
|
| 144 |
+
"use_liger_kernel": false,
|
| 145 |
+
"liger_kernel_config": null,
|
| 146 |
+
"eval_use_gather_object": false,
|
| 147 |
+
"average_tokens_across_devices": true
|
| 148 |
+
}
|
training_loss_history.csv
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
loss,step
|
| 2 |
+
6.4565,500
|
| 3 |
+
4.3201,1000
|
| 4 |
+
4.2592,1500
|
| 5 |
+
4.2331,2000
|
| 6 |
+
4.2168,2500
|
| 7 |
+
4.2023,3000
|
| 8 |
+
4.1945,3500
|
| 9 |
+
4.1855,4000
|