Update README.md
Browse files
README.md
CHANGED
|
@@ -22,32 +22,32 @@ Running under the **JiRackTrain** pipeline on the enterprise infrastructure clus
|
|
| 22 |
## Model Details
|
| 23 |
|
| 24 |
- **Model Name**: TernaryTransformer3B (3.3 Billion Parameters)
|
| 25 |
-
- **Architecture**: TernaryTransformer (custom ternary bit-response logic
|
| 26 |
-
- **Tokenizer**: [JiRack Pro Tokenizer (128K)](https://huggingface.co/CMSManhattan/JiRack-Pro-Tokenizer-128K)
|
| 27 |
-
- 347 active language editions of Wikipedia
|
| 28 |
-
- Specialized robotic action tokens + traditional text/vision embeddings
|
| 29 |
- **Pre-training Dataset**: [JiRack-Pretrain-Dataset](https://huggingface.co/datasets/CMSManhattan/JiRack-Pretrain-Dataset)
|
| 30 |
-
- **Training Objective**: From-scratch pre-training using the exact same multimodal corpus used to optimize the JiRack Pro Tokenizer (perfect vocabulary alignment)
|
| 31 |
|
| 32 |
-
##
|
| 33 |
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
-
|
| 37 |
|
| 38 |
-
|
| 39 |
|
| 40 |
-
|
| 41 |
-
- **Data Layout**: Sequential pipeline distributed across **7 cloud shards** (`jirack_pretrain_chunk_0.pt` through `jirack_pretrain_chunk_6.pt`)
|
| 42 |
-
- **Infrastructure**: Enterprise cluster with network-mounted storage (`/mnt/nfs_clientshare/JiRackTrain`)
|
| 43 |
-
- **Framework**: PyTorch + Accelerate (Python 3.12)
|
| 44 |
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
-
|
| 48 |
-
Shards found for processing: 7
|
| 49 |
-
Initializing JiRack architecture (3.3B) from your class...
|
| 50 |
-
-> Gradient Checkpointing successfully activated on TernaryTransformer3B.
|
| 51 |
-
Starting sequential training loop across shards...
|
| 52 |
|
| 53 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
## Model Details
|
| 23 |
|
| 24 |
- **Model Name**: TernaryTransformer3B (3.3 Billion Parameters)
|
| 25 |
+
- **Architecture**: TernaryTransformer (custom ternary bit-response logic)
|
| 26 |
+
- **Tokenizer**: [JiRack Pro Tokenizer (128K)](https://huggingface.co/CMSManhattan/JiRack-Pro-Tokenizer-128K) — 347 active language editions of Wikipedia + specialized robotic action tokens
|
|
|
|
|
|
|
| 27 |
- **Pre-training Dataset**: [JiRack-Pretrain-Dataset](https://huggingface.co/datasets/CMSManhattan/JiRack-Pretrain-Dataset)
|
|
|
|
| 28 |
|
| 29 |
+
## Key Metrics (Step ~838/1000 on Shard 1)
|
| 30 |
|
| 31 |
+
- **Iteration Speed**: ~242–352 seconds per iteration (network tensor loading)
|
| 32 |
+
- **Current Loss**: 2.9678
|
| 33 |
+
- **Moving Average Loss**: 6.0920
|
| 34 |
+
- **Perplexity (PPL)**: 442.33
|
| 35 |
|
| 36 |
+
Engineers anticipate a **steep drop** in the perplexity curve within the first 250 iterations of Shard 1 as the ternary weights align with the tokenizer's token distributions.
|
| 37 |
|
| 38 |
+
## Intended Use
|
| 39 |
|
| 40 |
+
This foundational model is being developed specifically for **robotics applications**:
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
+
- Real-time control policies
|
| 43 |
+
- Multimodal reasoning (text + vision + action tokens)
|
| 44 |
+
- Edge deployment with ternary efficiency
|
| 45 |
+
- Low-latency physical interaction loops
|
| 46 |
|
| 47 |
+
## Monitoring & Updates
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
+
Training is actively progressing across all **7 shards**. Follow this repository or the linked tokenizer/dataset cards for checkpoint releases, evaluation results, and fine-tuned robotic variants.
|
| 50 |
+
|
| 51 |
+
---
|
| 52 |
+
|
| 53 |
+
**Stay tuned** — the ternary robotics revolution is just getting started! 🚀
|