[RFC] Visualizing Relay program as graph

Development pre-RFC
#21

Do we consider adding this support on NETRON? GitHub - lutzroeder/netron: Visualizer for neural network, deep learning, and machine learning models

2 Likes
#22

Yes I though about Netron but… unfortunately I am too bad at JS :frowning:

Back then I was thinking relay.as_json() and Netron consume that .json (but we still need a relay.js or so to interpret that JSON.)

Let me think about this.

#23

@chiwwang There is some similar work here:

Would this be of any help?

#24

any way to visualize optimized relay IR? this seems not working to visualize optimized relay IR.

opt_level = 4
target = tvm.target.Target('cuda')
with tvm.transform.PassContext(opt_level=opt_level):
    mod,params = tvm.relay.optimize(mod,target,params=params)
# visualize mod['main']
1 Like
#25

I think Netron is the most popular one.