teddyoas commited on
Commit
6ed52f3
·
1 Parent(s): 372381b

Update model card for promoted 1-epoch weights

Browse files
Files changed (1) hide show
  1. README.md +27 -20
README.md CHANGED
@@ -14,7 +14,27 @@ pipeline_tag: text-generation
14
 
15
  # Agron (domain-pretrained: programming languages + cybersecurity)
16
 
17
- Agron continued pretraining started from [`teddyoas/Agron`](https://huggingface.co/teddyoas/Agron) (published checkpoint @ commit [`a311e94`](https://huggingface.co/teddyoas/Agron/commit/a311e94867629620de8d0a45136acb5d0ba91755), ~50.3M params LLaMA-style).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 | 6.740 / 845.3 | 1.453 / 4.28 | -5.287 |
44
- | cyber | 7.632 / 2063.7 | 1.358 / 3.89 | -6.274 |
45
- | replay (wikitext) | 7.208 / 1350.8 | 5.626 / 277.6 | -1.582 |
46
 
47
  ![Domain pretraining curves](domain_pretraining_curve.png)
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") # this repo updated with these weights
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
  ![Domain pretraining curves](domain_pretraining_curve.png)
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`