Advertisement

PyTorch:调试源代码(Python深度学习)

阅读量:

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

  • Getting Started Debugging PyTorch
    • A Program To Debug
    • How To Debug

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'
    ,

全部评论 (0)

还没有任何评论哟~