Pytorch implementation of arbitrary learning rate and momentum schedules, including the One Cycle Policy - View it on GitHub