不同于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上,。

效果