WeakTr
WeakTr copied to clipboard
训练二分类数据集,attn_loss 停留在0.6934 (0.6933)
作者你好!我下载了你的代码,用自己的数据集进行训练,自己的数据集仅包含一个类别,含有具有该类别的图像以及不含有该类别的图像,训练过程中attn_loss一直无法收敛,一直停留再0.6933,请问一下可以修改哪些参数有助于attn_loss的收敛。
我用训练好的模型(attn_loss未收敛),打印了图像models.py中fine_cam,其值都很小,但可视化的图像是正常的
torch.sigmoid(fine_cam_pred)的值为0.5