Skip to content

Instantly share code, notes, and snippets.

View hyqneuron's full-sized avatar

Hou Yunqing hyqneuron

View GitHub Profile
@hyqneuron
hyqneuron / pytorch_visualize.py
Created June 7, 2017 07:06
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.