Transformer Infra Lab 实现记录

53 words

1 minute

2026-10-21

No Tags

CS336 Assignment 1#

文件树

1
- cs336-a1
2
  - cs336_basics
3
    - train_bpe.py        # 从原始文本学出词表合并规则
4
  - tests
5
    - adapters.py         # 自己写的代码和官方测试的接口

测试

1
cd cs336-a1 && uv run pytest tests/test_train_bpe.py -q

train_bpe

Transformer Infra Lab 实现记录

Author

zlxFTH

Published at

2026-10-21

License