Official code for the paper "MASAGA: A Linearly-Convergent Stochastic First-Order Method for Optimization on Manifolds". We obtained the projection definitions from "manopt".
- Pytorch version 0.4 or higher.
python main.py -e synthetic_L -m train -r 1
python main.py -e ocean_L -m train -r 1
python main.py -e mnist_L -m train -r 1
If you find the code useful for your research, please cite:
@article{babanezhadmasaga,
title={MASAGA: A Linearly-Convergent Stochastic First-Order Method for Optimization on Manifolds},
author={Babanezhad, Reza and Laradji, Issam H and Shafaei, Alireza and Schmidt, Mark},
journal={ECML},
year={2018}
}