pytorch代码调试工具

https://github.com/zasdfgbnm/TorchSnooper

pip install torchsnooper

在函数前加装饰器@torchsnooper.snoop()