Skip to content

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

@iefengjian

Description

@iefengjian

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions