迁移模型运行手写数字加法机(MNIST)
# 导入所需要的包,请保证torchvision已经在你的环境中安装好 import torch import torch.nn as nn from torch.autograd import Variable import torch.optim as optim import torch.nn.functional as F import torchvision.datasets as dsets import torchvision.transforms as transforms import matplotlib.pyplot as plt import numpy as np import copy %matplotlib inline View Code # 加载数据+随机处理+两组不同MNIST数据集 # 设置图像读取器的超参数 image_size = 28 # 图像的总尺寸28*28 num_classes = 10 # 标签的种类数 num_epochs = 20 # 训练的总循环周期 batch_size = 64 # 批处理的尺寸大小 # 如果系统中存在着GPU,我们将用GPU来完成张量的计算 use_cuda = torch.cuda.is_available() # 定义一个布尔型变量,标志当前的GPU是否可用 # 如果当前GPU可用