GitHub repository: timgaripov/swa
Primary language: Python
Description: Stochastic Weight Averaging in PyTorch