2024最新Python Debugger工具pdb的用法(深度学习项目),了解输入输出的形状大小
侵入式方法 (在被调试的代码中添加以下代码然后再正常运行代码)
import pdb
pdb.set_trace()
例如:
正常运行训练文件后:
在命令行发现输出以下内容:
出现了(Pdb) 的提示符,说明已经打开pdb
在使用Python的pdb调试器时,如果你想知道某个变量或表达式的结果大小,你可以在调试器中直接打印这个变量。
在上面的代码中,如果你想要知道F.max_pool2d(self.encoder2(out), 2, 2)
操作后的结果大小,你可以在pdb调试器的命令行中输入以下命令:
print(F.max_pool2d(self.encoder2(out), 2, 2).shape)
结果如下:
基于以上步骤,你就可以知道每个输出结果的形状大小!