为以下代码添加注释# Download training data from open datasets. training_data = datasets.FashionMNIST( root="data", train=True, download=True, transform=ToTensor(), ) # Download test data from open datasets. test_data = datasets.FashionMNIST( root="data", train=False, download=True, transform=ToTensor(), )
时间: 2023-07-14 12:12:47 浏览: 171
tf.data.Dataset.map与tf.data.Dataset.interleave
# 导入 PyTorch 内置的 FashionMNIST 数据集
from torchvision import datasets
# 导入 PyTorch 内置的 Transforms 转换模块
from torchvision.transforms import ToTensor
# 从 open datasets 下载并加载训练数据集
training_data = datasets.FashionMNIST(
root="data", # 数据集存放路径
train=True, # 表示是训练集,因此会从该路径下找到训练数据
download=True, # 如果数据集不存在,就先下载
transform=ToTensor(), # 将数据转换为 Tensor 类型
)
# 从 open datasets 下载并加载测试数据集
test_data = datasets.FashionMNIST(
root="data", # 数据集存放路径
train=False, # 表示是测试集,因此会从该路径下找到测试数据
download=True, # 如果数据集不存在,就先下载
transform=ToTensor(), # 将数据转换为 Tensor 类型
)
#注释解释了代码的作用,参数的含义以及每个参数的用途。
阅读全文