神经网络模型可视化工具包

比较好用的一个可视化网络的工具，使用起来Bug比较少。

 from torchviz import make_dot
 #模型的超参数
dummy_input =  {
    "x_ops_1": X_ops_1,
    "x_adj_1": X_adj_1.to(torch.long),
    "x_ops_2":  X_ops_1,
    "x_adj_2": X_adj_1.to(torch.long),
    "zcp": zcp,
    "norm_w_d": norm_w_d_1,
    "hw_idx": hw_idx
}
y = model(**dummy_input) # your model
dot = make_dot(y, params=dict(model.named_parameters()))
dot.format = 'png'
dot.render('GIN_Model_structure') # 本地生成的模型图

之后会在本地环境下产生一个名为GIN_Model_structure.png网络的结构图。