search

torrvision / crayon

A language-agnostic interface to TensorBoard
MIT License
779 stars 59 forks source link

Missing graph visualisation #22

Open edran opened 7 years ago

edran commented 7 years ago

We should write an interface for visualising PyTorch (or any other NN graph).

ytian81 commented 7 years ago

Any updates?

sunshineatnoon commented 7 years ago

Any updates? That's a cool function.

nashory commented 7 years ago

Any updates...?

albanD commented 7 years ago

Some experiments have been done to visualize Pytorch graphs in tensorflow using ONNX. The problem is that the graphs usually end up being too big because they print at the Function level. A way to add higher level scopes for nn Modules for example would be needed to get useful graphs.