Can't visualize operations in pytorch modules

parrt / tensor-sensor

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

https://github.com/parrt/tensor-sensor

MIT License

771 stars 39 forks source link

Can't visualize operations in pytorch modules #21

Open turian opened 3 years ago

turian commented 3 years ago

I'd like to see what is happening with my torch module, and it's submodules. But I only get the toplevel:

parrt commented 3 years ago

hi. it only does one level. it doesn't pull the code from voice(). can you add tsensor stuff to that func?