莫烦pytorch教程的CNN代码的一些旧版的修改
作为一个代码小白,最近在学习莫烦的pytorch教程,因为时间比较长了,有些地方需要做一些修改,写个日记记录下我的笔记。
导入包和数据预处理
import torch
import torch.nn as nn
import torch.utils.data as Data
import torchvision
import matplotlib.pyplot as plt
torch.manual_seed(1)
#超参数设置
EPOCH =