pytorch 模型可视化的例子

pytorch 模型可视化的例子

如下所示: 一. visualize.py from graphviz import Digraphimport torchfrom torch.autograd import Variable def make_dot(var, params=None): """ Produces Graphviz representation of PyTorch autograd gr...