src.loss_func

Classes

CrossEntropyLoss

Multi-class cross entropy loss between two permutations.

FocalLoss

Focal loss between two permutations.

HammingLoss

Hamming loss between two permutations.

InnerProductLoss

Inner product loss for self-supervised problems.

OffsetLoss

OffsetLoss Criterion computes a robust loss function based on image pixel offset.

PermutationLoss

Binary cross entropy loss between two permutations, also known as “permutation loss”.

PermutationLossHung

Binary cross entropy loss between two permutations with Hungarian attention.