⚠️注意:PyTorch 代码中使用 ipdb 调试方式,不支持多线程。如果有多个 worker,将 worker 设置为 0 即可。
ipdb 介绍
说明文档:ipdb
ipdb 调试器具有语法高亮显示、更好的回溯和更好的内省功能,与pdb模块具有相同的接口。
使用方法:
安装 ipdb:
pip install ipdb
快速使用:
直接在代码里需要调试的地方,另起一行,写入:
import ipdb; ipdb.set_trace()
就可以设置一个断点,程序会在 pdb.set_trace() 位置暂停,并进入 ipdb 调试环境。
举例:
demo.py
文件为:<