Skip to content

optyang/proxsgd

Repository files navigation

ProxSGD

This is the TensorFlow 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.1 and 4.3. The simulations in Section 4.2 are implemented in PyTorch and available at here.

About

ProxSGD algorithm in TensorFlow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages