Skip to content

Instantly share code, notes, and snippets.

@peroon

peroon/mxnet visualize graph.py

Created April 23, 2017 14:26
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save peroon/bcb938338eac8905221bf5340be8f4f9 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save peroon/bcb938338eac8905221bf5340be8f4f9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
mxnet visualize graph.py
# -*- coding: utf-8 -*-
import mxnet as mx
import numpy as np
import mxlib
if __name__ == '__main__':
# get model
mxlib.get_model('http://data.mxnet.io/models/imagenet/resnext/101-layers/resnext-101', 0)
sym, arg_params, aux_params = mx.model.load_checkpoint('resnext-101', 0)
# visualize
a = mx.visualization.plot_network(sym)
a.render('graphviz_graph')
# save to current directory
@peroon

peroon commented Apr 23, 2017

Copy link
Copy Markdown
Author

Before execute, install graphviz and set PATH (Windows)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment