Debug PyTorch code using PySnooper