训练siammask模型iou一直等于0

Question

最近在复现SiamMask项目，在训练siammask_base模块和siammask_refine模块时，mean_iou,iou_5,iou_7都是一直为0，调整学习率lr也没有用，不知道怎么解决。

Answer 1

pytorch的版本更新导致的，作者这个库的版本是0.4几，现在都1.几了，把siammask.py模型下的mask_sum = pred.eq(1).int().add(label.eq(1)变成mask_sum = pred.eq(1).int().add(label.eq(1).int())就ok了