PyTorch - 36 - PyTorch DataLoader源代码 - Debugging Session
Short Program To Debug PyTorch Source
在开始调试之前,我们只想简要介绍一下我们编写的程序,使我们可以进入并查看数据集的规范化,并确切地了解如何在引擎盖和PyTorch下完成它。
import torch
import torchvision
import torchvision.transforms as transforms
import torch.nn as nn
from torch.utils.data import DataLoader
从torch.utils.data导入DataLoader
正如我们在上一集中讨论的那样,我们具有均值和标准差值。现在,不必计算它们,我们只需要提取它们并将它们硬编码到此处的程序中即可。
mean = 0.2860347330570221
std = 0.3530242443084717
如果要脱机获取这些值,我们将做这种事情。
我们不想麻烦重新计算这些值,因此我们在这里很难对其进行描述。我们有均值和标准差,我们知道我们需要这两个值才能对数据集的每个成员或每个像素进行归一化。
接下来,我们使用FashionMNIST类构造函数初始化训练集。这里要注意或要注意的关键是变换。我们有变换的组成。
train_set = torchvision.datasets.FashionMNIST(
root='./data'
,train=True
,download