search

szagoruyko / pytorchviz

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

remove all jit trace related code and examples #54

Closed albanD closed 3 years ago

albanD commented 3 years ago

This is just a proposal to see if we're happy to go this way, or if we prefer to fix this by duplicating what is done in torch.utils.tensorboard.

The update to the notebook is simply removing the part that was showing the jit trace printing.

szagoruyko commented 3 years ago

thanks Alban!