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

操作步驟

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

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

2 . 使用如下代碼，保存在統(tǒng)計(jì)目錄下的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因?yàn)槲业哪Ｐ驮趃pu上。

效果