explain() likely affects PyTorch autograd operations

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

explain() likely affects PyTorch autograd operations #14

Open parrt opened 3 years ago

parrt commented 3 years ago

PyTorch: use with torch.no_grad() to prevent messing up autograd computations.