深度学习网络通常具有很深的层次结构,而且层与层之间通常会有并联、串联等连接方式。当使用PyTorch建立一个深度学习网络并输出文本向读者展示网络的连接方式是非常低效的,所以需要有效的工具将建立的深度学习网络结构有层次化的展示,这就需要使用相关的深度学习网络结构可视化库。
1 准备网络
import torch
import torch.nn as nn
import torchvision
import torchvision.utils as vutils
from torch.optim import SGD
import torch.utils.data as Data
from sklearn.metrics import accuracy_score
import matplotlib.pyplot as plt
"""
下面导入手写字体数据,并将数据处理为数据加载器
"""
train_data=torchvision.datasets.MNIST(
root="./Dataset",
train=True,
#将数据转化为torch使用的张量,取值范围为[0-1]
transform=torchvision.transforms.ToTensor(),
download=False#因为数据已经下载过所以这里不在下载
)
#定义一个数据加载器
train_loader=Data.DataLoader(
dataset=train_data,#使用的数据集
batch_size=128,#批处理样本大小
shuffle=True,
num_workers=0
)
#准备需要的测试集
test_data=torchvision.datasets.MNIST(
root="./Dataset",
train=Fa