Views
No views yet
Nwna/olmo3-190m-zh-v2-base,继续注入 Wikipedia-zh + Cosmopedia-Chinese + base replay 组成的 1.194B tokens 中文语料。这是《零基础 AI 大模型研发训练营》第 04 讲进阶作业模型。它是 continue-pretrained base model,不是 instruction/chat model。
11. 是否能从 base checkpoint 正确继续训练,而不是重新随机初始化。
22. tokenizer 是否和 base 模型严格一致。
33. Wiki / Cosmopedia 是否能提升事实密度和解释型表达。
44. continue pretrain 是否能替代 SFT。Nwna/olmo3-190m-zh-v2-base 相同:| 字段 | 值 |
|---|---|
architectures | ["Olmo3ForCausalLM"] |
hidden_size | 768 |
num_hidden_layers | 12 |
num_attention_heads | 12 |
intermediate_size | 3072 |
vocab_size | 48,000 |
max_position_embeddings | 4096 |
| training seq_len | 2048 |
| total params | ~187M |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tok = AutoTokenizer.from_pretrained("Nwna/olmo3-190m-zh-v2-continue")
4model = AutoModelForCausalLM.from_pretrained("Nwna/olmo3-190m-zh-v2-continue")
5
6inputs = tok("人工智能是", return_tensors="pt")
7outputs = model.generate(
8 **inputs,
9 max_new_tokens=120,
10 do_sample=True,
11 temperature=0.8,
12 top_p=0.9,
13)
14print(tok.decode(outputs[0], skip_special_tokens=True))1from transformers import AutoModelForCausalLM
2
3model = AutoModelForCausalLM.from_pretrained("Nwna/olmo3-190m-zh-v2-continue")
4# ... 用 SFTTrainer / 自定义 Trainer 做监督微调| 数据源 | 来源 | tokens | 作用 |
|---|---|---|---|
| base replay | base raw 抽样 | 300M | 防止灾难性遗忘,保留通用中文能力 |
| Wikipedia-zh | wikimedia/wikipedia config 20231101.zh | 394M | 注入实体、地点、历史、文学等百科事实 |
| Cosmopedia-Chinese | opencsg/chinese-cosmopedia | 500M | 注入解释型、教材型、知识组织文本 |
| 合计 | - | 1.194B | continue pretrain full data |
Nwna/olmo3-190m-zh-v2-tokenizer 重新编码,没有直接复用其他模型或数据集中的 tokenized .bin。这是必须条件:token id 的语义由 tokenizer 决定,混用 tokenizer 会导致 embedding / lm_head 语义错位。1actual_tokens: 1,194,000,000
2documents: 1,203,778
3file_size_bytes: 2,388,000,000
4max_token_id: 47,999
5eos_token_id: 0
6avg_traditional_probe_ratio: 0.000508
7noise_pattern_hit: false北京大学、四大发明、造纸术、印刷术、火药、指南针、红楼梦、贾宝玉、林黛玉、Python、人工智能 等关键词。抽查只能证明语料中有覆盖,不能保证模型一定稳定回答。| 项目 | 值 |
|---|---|
| base checkpoint | Nwna/olmo3-190m-zh-v2-base |
| GPU | H100 |
| precision | bf16 |
| attention | SDPA |
| epochs | 1.0 |
| total steps | 4,328 |
| per-device batch | 16 |
| grad accumulation | 8 |
| effective batch | 128 sequences |
| tokens / step | 262,144 |
| peak lr | 2e-4 |
| min lr | 2e-5 |
| warmup ratio | 10% |
| scheduler | warmup + cosine |
| optimizer | AdamW |
| weight decay | 0.1,embedding / norm / bias excluded |
| steps | train_loss | eval_loss | GPU |
|---|---|---|---|
| 20 | 3.4235 | 3.3283 | A100-80G |
| metric | value |
|---|---|
| total steps | 4,328 |
| train_runtime | 8,964.52s |
| train_steps_per_second | 0.483 |
| train_loss | 3.0431 |
| last_eval_loss | 2.9404 |
| final learning_rate | 2.0002e-05 |
| final grad_norm | 0.2779 |
src/eval.py 的采样生成设置:1max_new_tokens=120
2do_sample=True
3temperature=0.8
4top_p=0.9
5repetition_penalty=1.1| Prompt | Base | Continue | 变化 |
|---|---|---|---|
| 人工智能是 | 🟡 可生成科普,但较泛 | 🟡 更像教材式分领域展开 | 风格增强 |
| 山里有座庙 | 🟡 偏文物/寺庙说明 | 🟡 更像故事长叙事 | 续写更长,但仍虚构 |
| 今天天气不错,我准备 | 🟡 对话关系混乱 | 🟡 叙事更完整但漂移到奇幻故事 | 表达增强,控制仍弱 |
| 北京大学位于 | 🔴 答错地点 | 🔴 仍答错地点 | 未解决事实问答 |
| 四大发明是 | 🔴 错到电力设备列表 | 🔴 错到科学史人物 | 未解决事实问答 |
| 《红楼梦》人物 | 🔴 标题式重复 | 🔴 输出过短,未答人物 | 未解决文学知识 |
| Python 是一种 | 🟡 知道与编程相关但定义混乱 | 🟡 更像解释型文本,但仍不准确 | 小幅改善 |
do_sample=True 时,同一 prompt 多次生成可能不同。ch04(下)进阶作业:持续预训练_v2.md1@misc{nwna-olmo3-190m-zh-v2-continue,
2 title={OLMo3-190M-zh-v2-continue: Chinese Continue-Pretrained Teaching Model},
3 author={Nwna},
4 year={2026},
5 howpublished={\url{https://huggingface.co/Nwna/olmo3-190m-zh-v2-continue}},
6 note={LLM001 Course, Lecture 04 advanced homework}
7}