Badnyal commited on
Commit
fdcba43
·
verified ·
1 Parent(s): 2f36c7f

Initial release of Meitei-RoBERTa-Base (Bengali Script) with training logs and model card.

Browse files
Files changed (42) hide show
  1. README.md +62 -0
  2. checkpoint-1380/config.json +26 -0
  3. checkpoint-1380/model.safetensors +3 -0
  4. checkpoint-1380/optimizer.pt +3 -0
  5. checkpoint-1380/rng_state.pth +3 -0
  6. checkpoint-1380/scaler.pt +3 -0
  7. checkpoint-1380/scheduler.pt +3 -0
  8. checkpoint-1380/special_tokens_map.json +37 -0
  9. checkpoint-1380/tokenizer.json +0 -0
  10. checkpoint-1380/tokenizer_config.json +53 -0
  11. checkpoint-1380/trainer_state.json +48 -0
  12. checkpoint-1380/training_args.bin +3 -0
  13. checkpoint-2760/config.json +26 -0
  14. checkpoint-2760/model.safetensors +3 -0
  15. checkpoint-2760/optimizer.pt +3 -0
  16. checkpoint-2760/rng_state.pth +3 -0
  17. checkpoint-2760/scaler.pt +3 -0
  18. checkpoint-2760/scheduler.pt +3 -0
  19. checkpoint-2760/special_tokens_map.json +37 -0
  20. checkpoint-2760/tokenizer.json +0 -0
  21. checkpoint-2760/tokenizer_config.json +53 -0
  22. checkpoint-2760/trainer_state.json +69 -0
  23. checkpoint-2760/training_args.bin +3 -0
  24. checkpoint-4140/config.json +26 -0
  25. checkpoint-4140/model.safetensors +3 -0
  26. checkpoint-4140/optimizer.pt +3 -0
  27. checkpoint-4140/rng_state.pth +3 -0
  28. checkpoint-4140/scaler.pt +3 -0
  29. checkpoint-4140/scheduler.pt +3 -0
  30. checkpoint-4140/special_tokens_map.json +37 -0
  31. checkpoint-4140/tokenizer.json +0 -0
  32. checkpoint-4140/tokenizer_config.json +53 -0
  33. checkpoint-4140/trainer_state.json +90 -0
  34. checkpoint-4140/training_args.bin +3 -0
  35. config.json +26 -0
  36. model.safetensors +3 -0
  37. special_tokens_map.json +37 -0
  38. tokenizer.json +0 -0
  39. tokenizer_config.json +53 -0
  40. training_args.bin +3 -0
  41. training_args.json +148 -0
  42. 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