- 安装graphviz
pip3 install graphviz --user
- 安装pytorchviz
pip3 install git+https://github.com/szagoruyko/pytorchviz --user
- 测试
from torchviz import make_dot
- 使用
from torchviz import make_dot
...
atts, dets = model(images)
vis_graph = make_dot(model(images), params=dict(model.named_parameters()))
vis_graph.view()
- 结果
- 参考资料
使用pytorchviz进行pytorch执行过程的可视化