Imported from https://github.com/201419/Optimizer-PyTorch