PyTorch - 20 - 如何调试PyTorch源代码 - Python中的深度学习

Getting Started Debugging PyTorch

当然，第一步是在系统上安装Visual Studio Code。 一旦完成，我们就可以开始了。

A Program To Debug

要进行调试，我们需要一个可以用来启动程序的程序。

我们在这里创建了一个示例程序，将用于调试一些PyTorch源代码。 我们将调试实际上位于torchvision软件包中的Fashion MNIST数据集。 但是，尽管如此，一切都会变得一样。

import torch
import torchvision
import torchvision.transforms as transforms

train_set = torchvision.datasets.FashionMNIST(
    root='./data'
    ,train=True
    ,download=True
    ,transform=transforms.Compose([
        transforms.ToTensor()
    ])
)

image, label = train_set[0]

print(image.shape)

How To Debug

为了在Visual Studio Code中调试任何python代码，我们需要安装python调试扩展。这将为我们提供调试功能。安装调试扩展后，请按照以下步骤操作。

1. 放置断点
2. 在调试模式下运行程序
3. 使用键盘手动控制程序执行
4. 走进PyTorch

请注意，还需要调试导入的库（例如PyTorch）。我们需要在VS Code的launch.json文件中将justMyCode属性设置为false。

以下是示例启动配置。

{
    
    
    "name": "Python: Current File",
    "type": "python",
    "request": "launch",
    "program": "${file}",
    "justMyCode": false,
    "cwd": "${fileDirname}"
}

调试代码是了解正在发生的事情的绝对最佳方法之一。请注意，我们无需编写大量代码即可实际设置调试PyTorch。在我们的示例中，只有15或16行代码。

要调试PyTorch源代码，我们只需要通向我们要调试的PyTorch代码的路径。