Imported from https://github.com/4uiiurz1/pytorch-adacos