Skip to content

Instantly share code, notes, and snippets.

Avatar

Hou Yunqing hyqneuron

View GitHub Profile
@hyqneuron
hyqneuron / pytorch_visualize.py
Created Jun 7, 2017
PyTorch graph visualization
View pytorch_visualize.py
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.