文件名称：ADMM-NeuralNetworks:训练无梯度的神经网络
文件大小：29.03MB
文件格式：ZIP
更新时间：2024-06-07 08:29:58
Python
ADMM-神经网络 训练没有梯度的神经网络：一种ADMM方法（具有急切执行力和Matplotlib的Matlab，Tensorflow1.6-Python2.7 / 3.5）注意：您不能将此代码用于任何分配或任何盈利产品。 该代码是根据以下论文实施的； 但是，我使用均方误差损失而不是二进制铰链损失 泰勒，加文等。 “训练没有梯度的神经网络：可扩展的admm方法。” 机器学习国际会议。 2016年。 仅支持GPU？ 我只为GPU版本实现并在MNIST数据上进行了测试。 怎么跑？ 运行main.m / main.py
【文件预览】：
ADMM-NeuralNetworks-master
----matlab()
--------argminz.m(1KB)
--------trainY.mat(50KB)
--------main.m(4KB)
--------lambda_update.m(312B)
--------relu.m(75B)
--------argminlastz.m(494B)
--------testY.mat(8KB)
--------argminlast_ez.m(507B)
--------argmin_ez.m(1013B)
--------activation_update.m(894B)
--------weight_update.m(626B)
--------trainX.mat(25.25MB)
--------testX.mat(4.2MB)
----README.md(614B)
----python()
--------ADMM_NN.py(10KB)
--------main.py(2KB)