文章目录CNNMNIST手写数据CNN模型RNN(分类)MNIST手写数据RNN模型训练RNN(回归)
CNN
MNIST手写数据
import torch
import torch.nn as nn
import torch.utils.data as Data
import torchvision
import matplotlib.pyplot as plt
EPOCH = 1
BATCH_SIZE = 50
LR = 0.001
DOWNLOAD_MNIST = False # 如果已经