← all repositories

CoinCheung/pytorch-loss

Collection of PyTorch loss functions and CUDA-accelerated operators for training deep learning models.

2.3k stars Python ML FrameworksComputer Vision
pytorch-loss
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of various loss functions used in deep learning training, including label smoothing, AMSoftmax, focal loss, triplet loss, dice loss, and IoU-based losses (GIOU/DIOU/CIOU). It also includes activation functions (Swish, Mish), one-hot encoding utilities with CUDA support, exponential moving average operators, and convolution variants like CoordConv and DynamicConvolution. Some operators are implemented as CUDA extensions requiring compilation.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.