Skip to content

Training Neural Networks Without Gradients: A ADMM Approach

Notifications You must be signed in to change notification settings

PotatoThanh/ADMM-NeuralNetworks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

ADMM-NeuralNetworks

Training Neural Networks Without Gradients: An ADMM Approach (Matlab, Tensorflow1.6-Python2.7/3.5 with eager execution and matplotlib) Note: You cannot use this code for any assignment or any profit product.

This code was implemented follow below paper; however, I used Mean Square Error loss instead of Binary Hinge loss

Taylor, Gavin, et al. "Training neural networks without gradients: A scalable admm approach." International Conference on Machine Learning. 2016.

Only support for GPU?

I only implemented for GPU version and tested on MNIST data.

How to run?

Run main.m/main.py

About

Training Neural Networks Without Gradients: A ADMM Approach

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published