search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Add open in colab badge #52

Closed szagoruyko closed 3 years ago

szagoruyko commented 3 years ago

Adds “open in colab badge” and adds a line to install the package in the first cell of examples.ipynb.

@albanD wdyt?