/proxsgd

ProxSGD

Primary LanguagePythonOtherNOASSERTION

ProxSGD

This is the PyTorch implementation of the ProxSGD algorithm proposed in following paper:

"ProxSGD: Training Structured Neural Networks under Regularization and Constraints," International Conference on Learning Representation (ICLR), Apr. 2020. URL

by Yang Yang, Yaxiong Yuan, Avraan Chatzimichailidis, Ruud JG van Sloun, Lei Lei, and Symeon Chatzinotas.

This repository contains the simulations in Section 4.2.