|
|
||
|---|---|---|
| .vscode | ||
| src | ||
| LANGCHAIN_LEARN.md | ||
| README.md | ||
| day1_char_rnn.py | ||
| requirements.txt | ||
README.md
从零训练一个迷你语言模型
第 1 天:字符级 RNN(GRU)
环境要求
- Python 3.10+
- PyTorch(Apple Silicon 支持 MPS)
安装与准备
- python -m venv .venv
- source .venv/bin/activate
- pip install torch
运行
- python day1_char_rnn.py
你应该看到
- 每 100 步输出一次 loss
- 每 200 步输出一次采样文本
说明
- 这是一个极小的数据集和模型,只为验证完整训练流程。
- 下一步我们会把硬编码文本替换为文件,并加入评估指标。