pdb:简单的python debug (调试pytorch源码)
#准备工作conda create -n pytorch python=3.7
conda activate pytorch
#准备工作conda install pytorch torchvision cpuonly -c pytorch
conda install pdb
调试的代码:
import torch
import torchvision
import torchvision.transforms as transforms
train_set = torchvision.datasets.CIFAR10(
root="./data&#