PyTorch】借助PyTorch简洁地实现对Fashion MNIST数据集进行分类和预测

【PyTorch】借助PyTorch简洁地实现对Fashion MNIST数据集进行分类和预测

在进行一下实验之前,需配置好基础环境。第一次下载Fashion MNIST数据集会失败,可以手动下载,放到相应位置。若过程中出现缺少的包,直接pip install 下载即可。
(1) 导入相关的包,主要是d2lzh_pytorch的导入。

import torch
import torchvision
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import time
import sys
sys.path.append('anaconda3/lib'

你可能感兴趣的:(联邦学习-隐私,人工智能,机器学习,深度学习)