PyTorch Implementation of Stochastic Gradient Descent with Warm Restarts

PyTorch Implementation of Stochastic Gradient Descent with Warm Restarts