Ademamix Python Packages

pytorch-optimizer

optimizer & lr scheduler & loss function collections in PyTorch

pylocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix