pytorch网络模型的可视化总结
方法一:自己下载graphviz程序,然后编写make_dot函数,然后进行调用。下载graphviz程序可以参考:文章一、文章二参考的demo如下:import torchfrom torch.autograd import Variableimport torch.nn as nnfrom graphviz import Digraphclass CNN(nn.Module): def __init__(self): super(CNN, self).__in
原创
2020-09-17 20:07:53 ·
564 阅读 ·
0 评论