aimhubio / aim

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.
https://aimstack.io
Apache License 2.0
4.94k stars 299 forks source link

Graphviz Support #2879

Open JesseFarebro opened 1 year ago

JesseFarebro commented 1 year ago

🚀 Feature

It would be really nice to upload a Graphviz graph in the DOT language which would be rendered as an svg (or on a canvas) in Aim. Given Aim already makes use of D3 an existing library could be used like: https://github.com/magjac/d3-graphviz

Motivation

Some structured data would be very useful to visualize as a graph, in my use-case I want to use this for visualizing hyper-parameter configurations.

This feature would be especially nice as graphviz doesn't ship wheels with the appropriate binaries so you can't render graphs without additionally installing the graphviz binaries. If Aim supported this feature it would provide an out of the box experience for visualizing graphs (Aim is responsible for rendering, no need to install the graphviz binaries).

Pitch

The pitch is either to have:

Then run.track would have the ability to accept a graphviz.Graph object which would upload the DOT representation to Aim and display it in an explorer.

vincrichard commented 2 months ago

+1, It would be great to be able to track a Graviz object, in my case torchviz output. Right now, I don't think there is a good way to visualise it in aim.