Update README.md
Browse files
README.md
CHANGED
|
@@ -70,11 +70,11 @@ Alo-70M-Base is based on a scaled-down LLaMA transformer architecture optimized
|
|
| 70 |
* **Hidden Dimension ($d_{model}$):** 512
|
| 71 |
* **Intermediate FFN Dimension:** 1408
|
| 72 |
* **Attention Mechanism:** Grouped-Query Attention (GQA) with 8 query heads and 4 key-value heads.
|
| 73 |
-
* **Positional Embeddings:** Rotary Position Embeddings (RoPE) with a base frequency
|
| 74 |
* **Activation Function:** SwiGLU (`silu`)
|
| 75 |
* **Word Embeddings:** Untied (`tie_word_embeddings = False`). The static lookup tables account for 48.6% (33.5M) of the total parameter budget.
|
| 76 |
* **Context Window:** 1024 tokens.
|
| 77 |
-
* **Tokenizer:** Custom Bengali Byte-Pair Encoding (BPE) tokenizer ([spitfire4794/beng_bpe](https://huggingface.co/spitfire4794/beng_bpe)) with a highly optimized vocabulary size of 32,768 (
|
| 78 |
|
| 79 |
## Pre-Training Data
|
| 80 |
The model was pre-trained on a curated, strictly Bengali corpus containing approximately **19.25 billion tokens**. The dataset is a blend of:
|
|
@@ -85,9 +85,9 @@ The model was pre-trained on a curated, strictly Bengali corpus containing appro
|
|
| 85 |
Alo-70M-Base was initialized with completely randomized weights (no transfer learning) and trained from scratch.
|
| 86 |
* **Hardware:** Kaggle Tensor Processing Unit (TPU) v5e-8
|
| 87 |
* **Training Time:** 12.9 hours (wall time) across two sequential runs.
|
| 88 |
-
* **Optimizer:** AdamW (`adam_torch_xla`) with
|
| 89 |
* **Weight Decay:** 0.1
|
| 90 |
-
* **Learning Rate Schedule:** Cosine Decay with a gradual warmup phase, peaking at
|
| 91 |
* **Precision & Optimization:** Mixed `bfloat16` (`bf16`) precision with gradient checkpointing enabled to maximize throughput.
|
| 92 |
|
| 93 |
## Evaluation Results
|
|
|
|
| 70 |
* **Hidden Dimension ($d_{model}$):** 512
|
| 71 |
* **Intermediate FFN Dimension:** 1408
|
| 72 |
* **Attention Mechanism:** Grouped-Query Attention (GQA) with 8 query heads and 4 key-value heads.
|
| 73 |
+
* **Positional Embeddings:** Rotary Position Embeddings (RoPE) with a base frequency theta of 10,000.
|
| 74 |
* **Activation Function:** SwiGLU (`silu`)
|
| 75 |
* **Word Embeddings:** Untied (`tie_word_embeddings = False`). The static lookup tables account for 48.6% (33.5M) of the total parameter budget.
|
| 76 |
* **Context Window:** 1024 tokens.
|
| 77 |
+
* **Tokenizer:** Custom Bengali Byte-Pair Encoding (BPE) tokenizer ([spitfire4794/beng_bpe](https://huggingface.co/spitfire4794/beng_bpe)) with a highly optimized vocabulary size of 32,768 (2^15) designed to maximize TPU efficiency.
|
| 78 |
|
| 79 |
## Pre-Training Data
|
| 80 |
The model was pre-trained on a curated, strictly Bengali corpus containing approximately **19.25 billion tokens**. The dataset is a blend of:
|
|
|
|
| 85 |
Alo-70M-Base was initialized with completely randomized weights (no transfer learning) and trained from scratch.
|
| 86 |
* **Hardware:** Kaggle Tensor Processing Unit (TPU) v5e-8
|
| 87 |
* **Training Time:** 12.9 hours (wall time) across two sequential runs.
|
| 88 |
+
* **Optimizer:** AdamW (`adam_torch_xla`) with beta_1 = 0.9, beta_2 = 0.95, epsilon = 10^-8
|
| 89 |
* **Weight Decay:** 0.1
|
| 90 |
+
* **Learning Rate Schedule:** Cosine Decay with a gradual warmup phase, peaking at 6 times 10^-4.
|
| 91 |
* **Precision & Optimization:** Mixed `bfloat16` (`bf16`) precision with gradient checkpointing enabled to maximize throughput.
|
| 92 |
|
| 93 |
## Evaluation Results
|