Ignore None node in .saved_tensors (It happens in some DGL models).

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

MIT License

3.24k stars 279 forks source link

Ignore None node in .saved_tensors (It happens in some DGL models). #68

Open lizeyan opened 3 years ago

lizeyan commented 3 years ago

For DGL, some operators in it (e.g., GSpMMBackward) have None in .save_tensors. In such cases, make_dot just crashes. So I ignore the Nones in .save_tensors.