喜悦雨

文章
7
资源
0
加入时间
2年10月21天

VSCODE调试控制台的使用——以pytorch下神经网络的加载模型前向传播为例前言查看加载模型的参数对模型的前向传播进行跟踪

最近在做模型的量化与量化后模型在arm平台上的迁移工作,为了保证量化后的模型在python下与在arm下运行的一致性(相同输入下,不同环境下的输出完全一致),需要对python下模型前向传播的每个操作的激活值、权重、输入的量化参数(scale与zero_point)、权重的量化参数(scale与zero_point)都能查看,在这个查看的过程中,发现VSCODE帮了大忙,因此记录一下VSCODE调试控制台的使用,便于自己与他人后续参考使用。