使用main进行heatmap热力图可视化报错

[Leezzzzz]

使用main进行heatmap热力图可视化报如下错误
File "C:\Users\LiZhentao\Desktop\YOLOv5-Lite-master\YOLOv5-Lite-master\scripts\Grad_Cam.py", line 251, in forward
prediction, logits, _ = self.model(img, augment=False)
ValueError: not enough values to unpack (expected 3, got 2)
模型使用的是v5lite-g.pt，target-layer为model_23_m_2_cv2

请问如何解决