[Python / PyTorch] debug backward()

发布时间:2023-12-31 17:00

问题描述

在自定义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

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号