[Python / PyTorch] debug backward()

问题描述

在自定义Loss的中，其backward()函数不支持在PyCharm中进行断点调试

因此需要以其他方式进行断点调试

解决方案

参考：Is there a way to debug the backward method of Function class?

使用pdb，在对应位置增加如下代码：

        import pdb
        pdb.set_trace()

具体pdb命令，参考：10分钟教程掌握Python调试器pdb