上传者: 38685455
|
上传时间: 2021-12-10 19:40:28
|
文件大小: 45KB
|
文件类型: -
莫烦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 = 1
BATCH_SIZE = 50
LR = 0.001
DOWNLOAD_MNIST = Fal