/cdw-cross-entropy-loss

Implementation of the Class Distance Weighted Cross-Entropy Loss in PyTorch.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers