Skip to content

Instantly share code, notes, and snippets.


Hou Yunqing hyqneuron

View GitHub Profile
hyqneuron /
Created Jun 7, 2017
PyTorch graph visualization
import torch
import torch.nn as nn
from torch.nn import Parameter
from torch.autograd import Variable, Function
from collections import defaultdict
import graphviz
This is a rather distorted implementation of graph visualization in PyTorch.