Skip to content

Double softmax #5

Description

@bkj

On this line you're applying a softmax to the similarities.

Then later you apply cross_entropy, which is a log softmax + NLL loss.

I think you probably want to remove the first softmax.

~ Ben

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