这是一个用于演示 Transformer 架构的仓库。 原视频链接:
请确保已安装以下依赖:
- type.ipynb: 主要内容,包含 Transformer 架构的演示代码和说明。
- test.ipynb: 用于实时拉取测试的相关代码或脚本。
-
克隆仓库到本地:
git clone https://github.com/buyaojinchang/transformer_test.git cd transformer_test -
安装依赖:
conda create -n transformer_test python=3.10 conda activate transformer_test pip install torch
-
打开
type.ipynb进行学习和运行。 -
使用
test.ipynb进行实时测试。