Loss Functions Python Packages

auraloss

Collection of audio-focused loss functions in PyTorch

210K 862 77

pytorch-optimizer

optimizer & lr scheduler & loss function collections in PyTorch

179K 406 38

dists-pytorch

IQA: Deep Image Structure and Texture Similarity Metric

78K 478 46

focal-loss

TensorFlow implementation of focal loss

26K 191 42

torch-log-wmse

logWMSE, an audio quality metric & loss function with support for digital silence target. Useful for training and evaluating audio source separation systems.

3K 46 1

torch-l1-snr

Variations of L1 SNR Loss function for training audio source separation machine learning models

2K 43 0

pylandscape

This project propose the loss landscape analysis as effective methodology to understand the robustness against natural perturbation of QNN.

960 3 0

torch-log-wmse-audio-quality

logWMSE, an audio quality metric & loss function with support for digital silence target. Useful for training and evaluating audio source separation systems.

925 46 1

iqa-pytorch

Comparison of IQA models in Perceptual Optimization

870 322 42

bokbokbok

Custom Loss Functions and Evaluation Metrics for XGBoost and LightGBM

773 39 7

pytorch-focalloss

PyTorch implementation of binary and multi-class focal loss functions

700 1 0

balanced-loss

Easy to use class balanced cross entropy and focal loss implementation for Pytorch

543 98 9

gedml

GeDML is an easy-to-use generalized deep metric learning library, which contains state-of-the-art deep metric learning algorithms and auxiliary modules to build end-to-end compute vision systems

531 36 3