不同于keras的可視化，keras.utils.plot_model。pytorch模型可視化比較復雜，此處使用pytorchviz。

操作步驟

1 . 命令行安裝pytorchviz（之前要安裝graphviz，這個我keras時已經(jīng)安裝）

pip install git+https://github.com/szagoruyko/pytorchviz

2 . 使用如下代碼，保存在統(tǒng)計目錄下的pdf文件中，并打開。

print(model)
g=make_dot(model(torch.rand(1,3,32,32).cuda()),params=dict(model.named_parameters()))
g.view()

其中，torch.rand(1,3,32,32).cuda()為偽造的數(shù)據(jù)，.cuda因為我的模型在gpu上。

效果