WeakTr icon indicating copy to clipboard operation
WeakTr copied to clipboard

训练二分类数据集,attn_loss 停留在0.6934 (0.6933)

Open iefengjian opened this issue 1 year ago • 0 comments

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

iefengjian avatar Jan 19 '24 10:01 iefengjian