[PyTorch小试牛刀]实战四·CNN实现逻辑回归对FashionMNIST数据集进行分类（使用GPU）

[PyTorch小试牛刀]实战四·CNN实现逻辑回归对FashionMNIST数据集进行分类（使用GPU） 内容还包括了网络模型参数的保存于加载。 数据集 下载地址 代码部分python import torch as t import torchvision as tv import numpy as np import time # 超参数 EPOCH = 5 BATCH_SIZE = 1

>>阅读原文<<