vscode调试python(transformers库的llama为例)

汕尾海湾  论坛元老 | 2025-4-7 15:50:30 | 来自手机 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 1766|帖子 1766|积分 5298

本文参考链接如下:
transformers源码阅读——vscode调试llama模型代码
1. 调试文件

因为只是搞清楚数据的流向,以是不需要下载任何预训练的参数
  1. from transformers.models.llama import LlamaModel,LlamaConfig
  2. import torch
  3. def run_llama():
  4.     # part 1  模型初始化
  5.     llamaconfig = LlamaConfig(vocab_size=32000,
  6.                               hidden_size=4096//2,
  7.                               intermediate_size=11008//2,
  8.                               num_hidden_layers=32//2,
  9.                               num_attention_heads=32//2,
  10.                               max_position_embeddings=2048//2)
  11.    
  12.     llamamodel = LlamaModel(config = llamaconfig)
  13.     # part 2 构建输入数据,在正向传播过程中,只需要传一个inputs_ids,batchsize是4,序列长度是30,每个元素是一个(0,llamaconfig.vocab_size)的随机整数
  14.     inputs_ids = torch.randint(
  15.         low=0,
  16.         high=llamaconfig.vocab_size,
  17.         size=(4,30))
  18.    
  19.     # part 3
  20.     res = llamamodel(inputs_ids)##前向计算,等价于llamamodel.forward(inputs_ids)
  21.     print(res)
  22. if __name__ == "__main__":
  23.     run_llama()
复制代码
2. vscode调试

2.1 设置调试设置文件

点击左侧边栏的运行和调试按钮:

然后点击创建launch.json文件

然后会出现 python文件调试当前正在运行的python文件:

然后就新建了一个launch.json调试设置文件

其中"justMyCode"表现是一个否只调试当前的文件,如果是true,就没法进入transformers的代码了,以是如果想调试进入transformers的代码,就把这个属性设置为false
2.2 设置断点

先看看怎么前向盘算的。按住ctrl键,点击LlamaModel进入LlamaModel界说的源码
但是LlamaModel界说的源码很多,可以通过资源管理器的大纲标签查看LlamaModel源码文件的代码布局:

为了快速抓住重点,在LlamaAttention的forward方法打断点:

调试窗口的断点标签可以找到这个断点,直接点击可以直接定位到断点的位置。

打开要调试的代码文件,然后点击调试窗口的调试按钮

然后步伐运行,在断点处停止运行。可以看到出现了调试按钮组件

在VSCode调试过程中,各操纵按钮的功能如下:

  • 继续(Continue)
    图标:通常为▶️(播放按钮)。
    功能:从当前停息的位置继续执行步伐,直到遇到下一个断点或步伐自然结束。
    使用场景:当检查完当前状态的变量或逻辑后,希望步伐继续运行而非逐行调试。
  • 逐过程(Step Over)
    图标:通常为⏭️(向右箭头跨越圆点)。
    功能:执行当前行的代码,不进入函数内部。若当前行包罗函数调用,会直接执行完该函数,并停在下一行。
    示例:在foo();这一行使用“逐过程”,会直接执行foo()函数全部代码,停在foo();的下一行。
  • 单步调试(Step Into)
    图标:通常为⏬(向下箭头进入函数)。
    功能:进入当前行代码中的函数调用内部,逐行调试函数内的逻辑。
    示例:在foo();这一行使用“单步调试”,会跳转到foo()函数的界说处,开始逐行执行函数内的代码。
  • 单步跳出(Step Out)
    图标:通常为⏫(向上箭头跳出函数)。
    功能:执行完当前函数的剩余代码,返回到调用此函数的位置
    使用场景:在函数内部调试时,若希望快速执行完当前函数并回到调用处。
  • 重启(Restart)
    图标:通常为

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

汕尾海湾

论坛元老
这个人很懒什么都没写!
快速回复 返回顶部 返回列表