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

最新推荐文章于 2024-06-20 21:23:56 发布

许喜远

最新推荐文章于 2024-06-20 21:23:56 发布

阅读量1.4k

点赞数 1

分类专栏： PyTorch 文章标签：深度学习 pytorch 机器学习数据挖掘神经网络

本文链接：https://blog.csdn.net/weixin_48367136/article/details/112532563

版权

本文介绍了如何在Visual Studio Code中调试PyTorch源代码，包括安装Visual Studio Code，创建调试程序，设置断点，以及调整launch.json配置以调试PyTorch库。通过这个教程，读者可以学习到如何更深入地理解PyTorch内部的工作原理。

摘要由CSDN通过智能技术生成

Getting Started Debugging PyTorch

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

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

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

import torch
import torchvision
import torchvision.transforms as transforms

train_set = torchvision.datasets.FashionMNIST

关注

专栏目录