赞
踩
前几周一直对如何调试某个类中的forward前向传播函数不知道如何操作,今天终于搞出来了,记录一下(之前调试函数时,即使在forward函数下打断点,但是调试时还是会直接进入类中的__init__ 函数...)
比如YOLOX代码,想看一下YOLOHead类中forward函数中的数据:
1. 首先找到类调用的地方,YOLOHead被YoloBody调用,YoloBody在trian.py上调用:
所以一定要在trian.py上进行debug按键,并且在此处不用打断点!
2. 在YoloBody类的forward函数下打断点,注意不要直接把断点打在函数或者类名上。因为self.backbone=YOLOPAFPN(),故运行到self.backbone.forward(x)语句时,会调用YOLOPAFPN类
3. 当程序运行到YOLOPAFPN类时,将断点打在forward函数下面的语句,此时点击调试键的Step into,不要点击Step Over,就可以进行forward函数的调用查看了!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。