https://github.com/feifeibear/admm-neuralnetwork
ADMM-NeuralNetwork was implemented by a potato
https://github.com/feifeibear/admm-neuralnetwork
Last synced: 4 months ago
JSON representation
ADMM-NeuralNetwork was implemented by a potato
- Host: GitHub
- URL: https://github.com/feifeibear/admm-neuralnetwork
- Owner: feifeibear
- Created: 2017-08-25T23:26:14.000Z (almost 9 years ago)
- Default Branch: master
- Last Pushed: 2017-06-24T03:54:02.000Z (about 9 years ago)
- Last Synced: 2025-01-23T00:41:17.038Z (over 1 year ago)
- Language: Matlab
- Homepage:
- Size: 28.3 MB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# ADMM-NeuralNetwork
ADMM-NeuralNetwork was implemented by a potato
I implemented ADMM-NeuralNetwork based on this paper: Taylor, Gavin, et al. "Training neural networks without gradients: A scalable admm approach." International Conference on Machine Learning. 2016.
# Requirements
- MATLAB version >= R2015a
- ADMM-NeuralNetwork only has GPU version; therefore, MATLAB and system have to support it.
# Neural Network Achitecture
- NN has 1 hidden layers. You can change it in NeuralNetwork.m and main.m
- I use Relu for activation function and least square for loss function.
# How to run
- Run main.m