Skip to content

BCELoss pos_weight parameter #67007

@emsidorov

Description

@emsidorov

Why BCEWithLogitsLoss has pos_weight and BCELoss doesn't?
And why they are different classes, doesn't it better to make one class for BCE loss with parameter from_logits=True/False?

cc @albanD @mruberry @jbschlosser @walterddr

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: nnRelated to torch.nntriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions