Model Graph可视化

看模型整体结构，不需要转onnx，直接看即可

1. 安装
apt-get install graphviz
pip install torchviz

2. 找到模型loss输出位置
from torchviz import make_dot
dot = make_dot(loss, params=dict(model.named_parameters()))
dot.render("graph", format="png") #输出模型结构图
``