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