Update model card for promoted 1-epoch weights
Browse files
README.md
CHANGED
|
@@ -14,7 +14,27 @@ pipeline_tag: text-generation
|
|
| 14 |
|
| 15 |
# Agron (domain-pretrained: programming languages + cybersecurity)
|
| 16 |
|
| 17 |
-
Agron
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
## Pretraining corpus
|
| 20 |
Mixed on-the-fly from flat token streams:
|
|
@@ -24,25 +44,13 @@ Mixed on-the-fly from flat token streams:
|
|
| 24 |
|
| 25 |
Corpus built by [`prepare_domain_corpus.py`](https://github.com/Njap-png/Agron/blob/main/prepare_domain_corpus.py).
|
| 26 |
|
| 27 |
-
## Training details
|
| 28 |
-
| Field | Value |
|
| 29 |
-
|---|---|
|
| 30 |
-
| Init weights | teddyoas/Agron (model.safetensors) |
|
| 31 |
-
| Steps | 5000 |
|
| 32 |
-
| Batch size | 16 × 1 grad acc = 16 |
|
| 33 |
-
| Sequence length | 512 |
|
| 34 |
-
| LR | 3e-4 (cosine, warmup 200) |
|
| 35 |
-
| Optimizer | AdamW (fused) |
|
| 36 |
-
| Tokens seen | ~41.0M |
|
| 37 |
-
| Hardware | 1× Tesla T4 (fp16) |
|
| 38 |
-
| Training script | [`pretrain_agron_domain.py`](https://github.com/Njap-png/Agron/blob/main/pretrain_agron_domain.py) |
|
| 39 |
-
|
| 40 |
## Held-out evaluation
|
|
|
|
| 41 |
| Source | Before (loss/PPL) | After (loss/PPL) | Δ |
|
| 42 |
|---|---|---|---|
|
| 43 |
-
| code |
|
| 44 |
-
| cyber |
|
| 45 |
-
| replay (wikitext) |
|
| 46 |
|
| 47 |

|
| 48 |
|
|
@@ -50,12 +58,11 @@ Corpus built by [`prepare_domain_corpus.py`](https://github.com/Njap-png/Agron/b
|
|
| 50 |
```python
|
| 51 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 52 |
tok = AutoTokenizer.from_pretrained("teddyoas/Agron")
|
| 53 |
-
model = AutoModelForCausalLM.from_pretrained("teddyoas/Agron") #
|
| 54 |
```
|
| 55 |
-
Or load from `teddyoas/Agron` directly (weights replaced by this upload).
|
| 56 |
|
| 57 |
## Files
|
| 58 |
- `model.safetensors`, `config.json`, `generation_config.json`
|
| 59 |
- `tokenizer.json`, `tokenizer_config.json`
|
| 60 |
- `domain_pretraining_curve.png`
|
| 61 |
-
- `eval_history.json`, `report.json`, `samples.json`
|
|
|
|
| 14 |
|
| 15 |
# Agron (domain-pretrained: programming languages + cybersecurity)
|
| 16 |
|
| 17 |
+
Agron is a ~50.3M-parameter LLaMA-style model. The weights in this repository are
|
| 18 |
+
the result of **one full epoch** of domain continued pretraining over code,
|
| 19 |
+
cybersecurity, and WikiText replay, started from the previously published Agron
|
| 20 |
+
checkpoint.
|
| 21 |
+
|
| 22 |
+
## Current promoted weights
|
| 23 |
+
|
| 24 |
+
| Field | Value |
|
| 25 |
+
|---|---|
|
| 26 |
+
| Steps | 7,246 (1 full epoch) |
|
| 27 |
+
| Tokens | 118.7M |
|
| 28 |
+
| Batch size | 16 x 2 grad accum = 32 |
|
| 29 |
+
| Sequence length | 512 |
|
| 30 |
+
| LR | 3e-4 (cosine, warmup 100) |
|
| 31 |
+
| Optimizer | AdamW (fused) |
|
| 32 |
+
| Hardware | 1x Tesla T4 (fp16) |
|
| 33 |
+
| Training time | ~61 min |
|
| 34 |
+
| Avg train loss | 1.319 |
|
| 35 |
+
|
| 36 |
+
Full metrics: [`report.json`](report.json), [`eval_history.json`](eval_history.json),
|
| 37 |
+
[`eval_before.json`](eval_before.json).
|
| 38 |
|
| 39 |
## Pretraining corpus
|
| 40 |
Mixed on-the-fly from flat token streams:
|
|
|
|
| 44 |
|
| 45 |
Corpus built by [`prepare_domain_corpus.py`](https://github.com/Njap-png/Agron/blob/main/prepare_domain_corpus.py).
|
| 46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
## Held-out evaluation
|
| 48 |
+
|
| 49 |
| Source | Before (loss/PPL) | After (loss/PPL) | Δ |
|
| 50 |
|---|---|---|---|
|
| 51 |
+
| code | 1.453 / 4.28 | 1.245 / 3.47 | -0.208 |
|
| 52 |
+
| cyber | 1.358 / 3.89 | 1.064 / 2.90 | -0.294 |
|
| 53 |
+
| replay (wikitext) | 5.626 / 277.6 | 5.024 / 152.0 | -0.602 |
|
| 54 |
|
| 55 |

|
| 56 |
|
|
|
|
| 58 |
```python
|
| 59 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 60 |
tok = AutoTokenizer.from_pretrained("teddyoas/Agron")
|
| 61 |
+
model = AutoModelForCausalLM.from_pretrained("teddyoas/Agron") # promoted 1-epoch weights
|
| 62 |
```
|
|
|
|
| 63 |
|
| 64 |
## Files
|
| 65 |
- `model.safetensors`, `config.json`, `generation_config.json`
|
| 66 |
- `tokenizer.json`, `tokenizer_config.json`
|
| 67 |
- `domain_pretraining_curve.png`
|
| 68 |
+
- `eval_history.json`, `eval_before.json`, `report.json`, `samples.json`
|