Skip to content

I think there is a wrong in Focalloss.py #16

@taylover-pei

Description

@taylover-pei

You write the focalloss just like this:
image
By doing this, the logpt is a scale not a tensor. If the batchsize is greater than 1, I think it is wrong here.
An I right?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions