spitfire4794 commited on
Commit
b8c4b39
·
verified ·
1 Parent(s): e8234f0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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 ($\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,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 $\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
 
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